🚀 Implement Tensor-type and basic methods #15
23
README.md
23
README.md
@ -1,22 +1,3 @@
|
|||||||
# Mainfold
|
# Manifold
|
||||||
|
|
||||||
```rust
|
A tensor implementation in Rust.
|
||||||
// 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]);
|
|
||||||
```
|
|
||||||
|
Loading…
Reference in New Issue
Block a user