コード例 #1
0
ファイル: common.py プロジェクト: yyaqi/Dragon 
 def LayerSetup(self, bottom):
     return _ops.Reshape(bottom, **self.arguments)
コード例 #2
0
def reshape(tensor, shape, name=None):

    return ops.Reshape(tensor, shape=shape, name=None)
コード例 #3
0
 def Setup(self, bottom):
     super(ReshapeLayer, self).Setup(bottom)
     input = bottom[0] if isinstance(bottom, list) else bottom
     return ops.Reshape(input, **self._param)