@ -3176,7 +3176,7 @@ class RecurrentLayerGroup(LayerBase):
@config_layer('pixel_softmax')
class PixelSoftmaxLayer(LayerBase):
def __init__(self, input, name, **xargs):
def __init__(self, name, inputs, **xargs):
super(PixelSoftmaxLayer, self).__init__(
name, 'pixel_softmax', 0, inputs=inputs, **xargs)
@ -126,6 +126,7 @@ __all__ = [
'row_conv_layer',
'dropout_layer',
'prelu_layer',
'pixel_softmax_layer',
]
@ -5905,8 +5906,8 @@ def pixel_softmax_layer(input, name=None, layer_attr=None):
else:
assert isinstance(input, collections.Sequence)
l = Layer(
inputs=[x.name for x in input],
name=name,
type=LayerType.PIXEL_SOFTMAX_LAYER,
**ExtraLayerAttribute.to_kwargs(layer_attr))
return LayerOutput(