Take references as input for Tensor::cat.

2025-06-19 11:56:45 +00:00 · 2023-06-25 13:03:05 +01:00
parent 5e03a1bc29
commit a9c113248a
2 changed files with 4 additions and 2 deletions
--- a/tests/tensor_tests.rs
+++ b/tests/tensor_tests.rs
@ -120,6 +120,7 @@ fn sum() -> Result<()> {
        tensor.sum(&[0])?.to_vec3::<u32>()?,
        &[[[5, 2, 11], [9, 7, 17]]],
    );
+    assert_eq!(tensor.sum(&[0, 2, 1])?.to_vec3::<u32>()?, &[[[51]]],);
    assert_eq!(
        tensor.t()?.sum(&[1])?.t()?.to_vec3::<u32>()?,
        &[[[8], [15]], [[10], [18]]]