We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3e95927 commit 6292bd2Copy full SHA for 6292bd2
cfpq_model/label_decomposed_graph.py
@@ -264,7 +264,7 @@ def mxm(
264
if rhs1 in self.matrices and rhs2 in other.matrices:
265
#if swap_operands:
266
mask = (accum.matrices[lhs].to_mask()
267
- if lhs in accum.matrices and other.matrices[rhs2].shape == self.matrices[rhs1].shape
+ if lhs in accum.matrices #and other.matrices[rhs2].shape == self.matrices[rhs1].shape
268
else None)
269
#else:
270
# mask = (self.matrices[lhs].to_mask()
0 commit comments