提交 670b5caa authored 作者: Adrian Seyboldt's avatar Adrian Seyboldt 提交者: Adrian Seyboldt

Fix tuple constructor in BaseCorrMM

上级 093665d9
......@@ -692,7 +692,7 @@ class CorrMM(BaseCorrMM):
if kern.type.ndim != 4:
raise TypeError("kern must be 4D tensor")
out_shape = tuple(
out_shape = (
1 if img.type.shape[0] == 1 else None,
1 if kern.type.shape[0] == 1 else None,
None,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论