manifold/README.md
Julius Koskela 27384a3044
Reduce cloning
Signed-off-by: Julius Koskela <julius.koskela@unikie.com>
2023-12-29 18:40:53 +02:00

655 B

Mainfold

// Create two tensors with different ranks and shapes
let mut tensor1 = Tensor::<i32, 2>::from([2, 2]); // 2x2 tensor
let mut tensor2 = Tensor::<i32, 1>::from([2]);    // 2-element vector

// Fill tensors with some values
tensor1.buffer_mut().copy_from_slice(&[1, 2, 3, 4]);
tensor2.buffer_mut().copy_from_slice(&[5, 6]);

// Calculate tensor product
let product = tensor1.tensor_product(&tensor2);

println!("T1 * T2 = {}", product);

// Check shape of the resulting tensor
assert_eq!(product.shape(), Shape::new([2, 2, 2]));

// Check buffer of the resulting tensor
assert_eq!(product.buffer(), &[5, 6, 10, 12, 15, 18, 20, 24]);