Trait dfdx::tensor::OnesTensor

pub trait OnesTensor<E>: Storage<E> {
    // Required method
    fn try_ones_like<S: HasShape>(
        &self,
        src: &S
    ) -> Result<Tensor<S::Shape, E, Self>, Self::Err>;

    // Provided methods
    fn ones<S: ConstShape>(&self) -> Tensor<S, E, Self> { ... }
    fn try_ones<S: ConstShape>(&self) -> Result<Tensor<S, E, Self>, Self::Err> { ... }
    fn ones_like<S: HasShape>(&self, src: &S) -> Tensor<S::Shape, E, Self> { ... }
}

Construct tensors filled with ones.

Required Methods

fn try_ones_like<S: HasShape>( &self, src: &S ) -> Result<Tensor<S::Shape, E, Self>, Self::Err>

Fallible version of OnesTensor::ones_like

Provided Methods

fn ones<S: ConstShape>(&self) -> Tensor<S, E, Self>

Creates a tensor filled with ones.

let a: Tensor<Rank2<2, 3>, f32, _> = dev.ones();

fn try_ones<S: ConstShape>(&self) -> Result<Tensor<S, E, Self>, Self::Err>

Fallible version of OnesTensor::ones

fn ones_like<S: HasShape>(&self, src: &S) -> Tensor<S::Shape, E, Self>

Build the tensor with a shape given by something else.

Given a shape directly:

let a: Tensor<(usize, Const<3>), f32, _> = dev.ones_like(&(5, Const));

Given another tensor:

let a: Tensor<Rank2<2, 3>, f32, _> = dev.ones();
let b: Tensor<_, f32, _> = dev.ones_like(&a);

Implementors

impl<E: Unit> OnesTensor<E> for Cpu

impl<E: Unit> OnesTensor<E> for Cudawhere Cpu: OnesTensor<E>,