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)()])