def LayerSetup(self, bottom):
return _ops.Reshape(bottom, **self.arguments)
def reshape(tensor, shape, name=None):
return ops.Reshape(tensor, shape=shape, name=None)
def Setup(self, bottom):
super(ReshapeLayer, self).Setup(bottom)
input = bottom[0] if isinstance(bottom, list) else bottom
return ops.Reshape(input, **self._param)