提交 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):
return Apply(
self,
(x, y, *new_inputs),
[
tensor(
dtype=x.type.dtype,
shape=tuple(1 if s == 1 else None for s in x.type.shape),
)
],
[x.type()],
)
def perform(self, node, inputs, out_):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论