pub trait Arange<E: Dtype>: Storage<E> + ZerosTensor<E> + TensorFromVec<E> {
// Provided method
fn arange<Size: Dim>(&self, n: Size) -> Tensor<(Size,), E, Self> { ... }
}
Expand description
Generates a tensor with ordered data from 0 to N
.
Provided Methods§
sourcefn arange<Size: Dim>(&self, n: Size) -> Tensor<(Size,), E, Self>
fn arange<Size: Dim>(&self, n: Size) -> Tensor<(Size,), E, Self>
Generates a tensor with ordered data from 0 to N
.
Const sized tensor:
let t: Tensor<Rank1<5>, f32, _> = dev.arange(Const::<5>);
assert_eq!(t.array(), [0.0, 1.0, 2.0, 3.0, 4.0]);
Runtime sized tensor:
let t: Tensor<(usize, ), f32, _> = dev.arange(5);
assert_eq!(t.as_vec(), [0.0, 1.0, 2.0, 3.0, 4.0]);