mirror of
https://github.com/gentoo-mirror/gentoo.git
synced 2026-08-23 18:38:08 -07:00
20 lines
649 B
Diff
20 lines
649 B
Diff
diff --git a/aesara/tensor/nnet/corr.py b/aesara/tensor/nnet/corr.py
|
|
index e89054d..77ed344 100644
|
|
--- a/aesara/tensor/nnet/corr.py
|
|
+++ b/aesara/tensor/nnet/corr.py
|
|
@@ -692,12 +692,12 @@ class CorrMM(BaseCorrMM):
|
|
if kern.type.ndim != 4:
|
|
raise TypeError("kern must be 4D tensor")
|
|
|
|
- out_shape = tuple(
|
|
+ out_shape = tuple([
|
|
1 if img.type.shape[0] == 1 else None,
|
|
1 if kern.type.shape[0] == 1 else None,
|
|
None,
|
|
None,
|
|
- )
|
|
+ ])
|
|
dtype = img.type.dtype
|
|
return Apply(self, [img, kern], [TensorType(dtype, shape=out_shape)()])
|
|
|