Trait burn::tensor::BroadcastArgs

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

Trait used for broadcast arguments.

Required Methods

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

Converts to a shape.

Implementations on Foreign Types

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

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

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

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

Implementors

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