Trait dfdx::tensor::ZerosTensor

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

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

Construct tensors filled with zeros.

Required Methods

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

Fallible version of ZerosTensor::zeros_like

Provided Methods

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

Creates a tensor filled with zeros.

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

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

Fallible version of ZerosTensor::zeros

fn zeros_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.zeros_like(&(5, Const));

Given another tensor:

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

Implementors

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

impl<E: Unit> ZerosTensor<E> for Cuda