Mentions légales du service

Skip to content
Snippets Groups Projects
Commit 4da99eb4 authored by hhakim's avatar hhakim
Browse files

Fix dimensions check in n-dimensional case of pyfaust.lazlinop.zeros mul.

parent 2c5218fd
No related branches found
No related tags found
No related merge requests found
......@@ -1527,7 +1527,7 @@ def zeros(shape):
"""
def _matmat(op, shape):
_sanitize_op(op)
if op.shape[0] != shape[1]:
if op.shape[-2] != shape[1]:
raise ValueError('Dimensions must agree')
if LazyLinearOp.isLazyLinearOp(op):
return zeros((shape[0], op.shape[1]))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment