Trait dfdx::data::Arange

pub trait Arange<E: Dtype>: Storage<E> + ZerosTensor<E> + TensorFromVec<E> {
    // Provided method
    fn arange<Size: Dim>(&self, n: Size) -> Tensor<(Size,), E, Self> { ... }
}

Generates a tensor with ordered data from 0 to N.

Provided Methods

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]);

Implementors

impl<E: Dtype, D: ZerosTensor<E> + TensorFromVec<E>> Arange<E> for D