未验证 提交 d4f03dfb 编写于 作者: X xiaoting 提交者: GitHub

fix for tuple,test=develop (#27190)

上级 91663073
......@@ -450,7 +450,7 @@ def interpolate(x,
for i in range(len(x.shape) - 2):
scale_list.append(scale)
attrs['scale'] = list(map(float, scale_list))
elif isinstance(scale, list) or isinstance(scale, float):
elif isinstance(scale, list) or isinstance(scale, tuple):
if len(scale) != len(x.shape) - 2:
raise ValueError("scale_shape length should be {} for "
"input {}-D tensor.".format(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册