We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6a5e3a8 commit 47fe93bCopy full SHA for 47fe93b
deep_causality_tensor/src/types/causal_tensor/op_tensor_ein_sum/ein_sum_impl.rs
@@ -388,7 +388,8 @@ where
388
389
let mut diag_sum = T::default();
390
for i in 0..diag_len {
391
- let flat_index = batch_offset + i * tensor.strides[axis1] + i * tensor.strides[axis2];
+ let flat_index =
392
+ batch_offset + i * tensor.strides[axis1] + i * tensor.strides[axis2];
393
diag_sum = diag_sum + tensor.data[flat_index].clone();
394
}
395
*res_val = diag_sum;
0 commit comments