We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents d8052c9 + 9fc4c26 commit dbed373Copy full SHA for dbed373
1 file changed
linear_operator/operators/block_linear_operator.py
@@ -97,7 +97,7 @@ def _getitem(self, row_index: IndexType, col_index: IndexType, *batch_indices: I
97
col_index = slice(col_start // num_blocks, col_end // num_blocks, None)
98
99
# Now we can try the super call!
100
- new_base_linear_op = self.base_linear_op._getitem(row_index, col_index, *batch_indices)
+ new_base_linear_op = self.base_linear_op._getitem(row_index, col_index, *batch_indices, _noop_index)
101
102
# Now construct a kernel with those indices
103
return self.__class__(new_base_linear_op, block_dim=-3)
0 commit comments