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