提交 f7c4e163 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Don't discard static shape information in AdvancedIncSubtensor

上级 e301a423
...@@ -3021,12 +3021,7 @@ class AdvancedIncSubtensor(Op): ...@@ -3021,12 +3021,7 @@ class AdvancedIncSubtensor(Op):
return Apply( return Apply(
self, self,
(x, y, *new_inputs), (x, y, *new_inputs),
[ [x.type()],
tensor(
dtype=x.type.dtype,
shape=tuple(1 if s == 1 else None for s in x.type.shape),
)
],
) )
def perform(self, node, inputs, out_): def perform(self, node, inputs, out_):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论