From fdd5b6891b00ffeb15cd48fe63f0e3517e8c89a8 Mon Sep 17 00:00:00 2001 From: Markus Mottl Date: Mon, 19 Jun 2017 14:40:44 -0400 Subject: [PATCH] Fixed dimension bug in outer function --- algopy/utpm/utpm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/algopy/utpm/utpm.py b/algopy/utpm/utpm.py index ddfafc7..80b0ba3 100644 --- a/algopy/utpm/utpm.py +++ b/algopy/utpm/utpm.py @@ -1976,7 +1976,7 @@ def outer(cls, x, y, out = None): assert x_shp[:2] == y_shp[:2] assert len(y_shp[2:]) == 1 - out_shp = x_shp + x_shp[-1:] + out_shp = x_shp + y_shp[-1:] out = cls(cls.__zeros__(out_shp, dtype = x.data.dtype)) cls._outer( x.data, y.data, out = out.data)