Trait dfdx::tensor_ops::Broadcast2

source · [−]

pub trait Broadcast2<T, const I1: isize, const I2: isize> {
    fn broadcast2(self) -> T;
}

Expand description

Broadcasts dimensions I1 and I2. Increases number dimensions by 2. Results in T.

Required Methods

fn broadcast2(self) -> T

Broadcast self into T, increasing number dimensions by 2.

Implementors

impl<const M: usize, const N: usize, H: Tape> Broadcast2<Tensor2D<M, N, H>, 0, 1> for Tensor0D<H>

impl<const M: usize, const N: usize, const O: usize, H: Tape> Broadcast2<Tensor3D<M, N, O, H>, 0, 1> for Tensor1D<O, H>

impl<const M: usize, const N: usize, const O: usize, H: Tape> Broadcast2<Tensor3D<M, N, O, H>, 0, 2> for Tensor1D<N, H>

impl<const M: usize, const N: usize, const O: usize, H: Tape> Broadcast2<Tensor3D<M, N, O, H>, 1, 2> for Tensor1D<M, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> Broadcast2<Tensor4D<M, N, O, P, H>, 0, 1> for Tensor2D<O, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> Broadcast2<Tensor4D<M, N, O, P, H>, 0, 2> for Tensor2D<N, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> Broadcast2<Tensor4D<M, N, O, P, H>, 0, 3> for Tensor2D<N, O, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> Broadcast2<Tensor4D<M, N, O, P, H>, 1, 2> for Tensor2D<M, P, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> Broadcast2<Tensor4D<M, N, O, P, H>, 1, 3> for Tensor2D<M, O, H>

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> Broadcast2<Tensor4D<M, N, O, P, H>, 2, 3> for Tensor2D<M, N, H>