assignment/
main.rs

1use tensor_compute::{s, GpuStore, Tensor};
2
3fn main() {
4    println!("Running in {:?}", GpuStore::get_default().info());
5    let mut ma = Tensor::from_data_and_shape(vec![1., 2., 3., 4., 5., 6., 7., 8.], vec![2, 2, 2]);
6    println!("{:?}", ma);
7    ma.assign(s!(1..2; ..; 1..2), 10.);
8    ma.assign(s!(0;0;0), -50.);
9    println!("{:?}", ma);
10    /*
11    Shape: [2, 2, 2]
12    [[[ -50  2 ]
13      [ 3  4 ]]
14
15     [[ 5  10 ]
16      [ 7  10 ]]]
17    */
18}