提交 ba4ab872 authored 作者: Ramana.S's avatar Ramana.S

Fix #3190

上级 dfb27303
...@@ -210,7 +210,7 @@ class DimShuffle(Op): ...@@ -210,7 +210,7 @@ class DimShuffle(Op):
ob.append(ib[value]) ob.append(ib[value])
output = TensorType(dtype=input.type.dtype, output = TensorType(dtype=input.type.dtype,
broadcastable=ob).make_variable() broadcastable=ob)()
return Apply(self, [input], [output]) return Apply(self, [input], [output])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论