Trait burn::tensor::BroadcastArgs

pub trait BroadcastArgs<const D1: usize, const D2: usize> {
    // Required method
    fn into_shape(self, shape: &Shape) -> Shape;
}

Trait used for broadcast arguments.

Required Methods

fn into_shape(self, shape: &Shape) -> Shape

Converts to a shape.

Implementations on Foreign Types

impl<const D1: usize, const D2: usize> BroadcastArgs<D1, D2> for [usize; D2]

fn into_shape(self, _shape: &Shape) -> Shape

impl<const D1: usize, const D2: usize, E> BroadcastArgs<D1, D2> for [E; D2]

where E: Element,

fn into_shape(self, shape: &Shape) -> Shape

Implementors

impl<const D1: usize, const D2: usize> BroadcastArgs<D1, D2> for Shape