Trait dfdx::tensor_ops::Reduce1

pub trait Reduce1<const I: isize>: Tensor<Dtype = f32> {
    type Reduced: Broadcast1<Self, I> + Tensor<Dtype = Self::Dtype, Tape = Self::Tape>;
    type DeviceR: Reduce1Axis<Self::Array, <Self::Reduced as HasArrayType>::Array, I>;
}

Reduce the Ith dimension of a Tensor. Enables functions like sum_axis() that reduce values along a single dimension.

Required Associated Types

type Reduced: Broadcast1<Self, I> + Tensor<Dtype = Self::Dtype, Tape = Self::Tape>

The resulting tensor type. The Ith dimension of this can be broadcast into Self via Broadcast1.

type DeviceR: Reduce1Axis<Self::Array, <Self::Reduced as HasArrayType>::Array, I>

Implementors

impl<H: Tape> Reduce1<-1> for Tensor0D<H>

type Reduced = Tensor0D<H>

type DeviceR = <Tensor0D<H> as HasDevice>::Device

impl<const M: usize, H: Tape> Reduce1<-1> for Tensor1D<M, H>

type Reduced = Tensor0D<H>

type DeviceR = <Tensor1D<M, H> as HasDevice>::Device

impl<const M: usize, const N: usize, H: Tape> Reduce1<0> for Tensor2D<M, N, H>

type Reduced = Tensor1D<N, H>

type DeviceR = <Tensor2D<M, N, H> as HasDevice>::Device

impl<const M: usize, const N: usize, H: Tape> Reduce1<-1> for Tensor2D<M, N, H>

type Reduced = Tensor1D<M, H>

type DeviceR = <Tensor2D<M, N, H> as HasDevice>::Device

impl<const M: usize, const N: usize, const O: usize, H: Tape> Reduce1<0> for Tensor3D<M, N, O, H>

type Reduced = Tensor2D<N, O, H>

type DeviceR = <Tensor3D<M, N, O, H> as HasDevice>::Device

impl<const M: usize, const N: usize, const O: usize, H: Tape> Reduce1<1> for Tensor3D<M, N, O, H>

type Reduced = Tensor2D<M, O, H>

type DeviceR = <Tensor3D<M, N, O, H> as HasDevice>::Device

impl<const M: usize, const N: usize, const O: usize, H: Tape> Reduce1<-1> for Tensor3D<M, N, O, H>

type Reduced = Tensor2D<M, N, H>

type DeviceR = <Tensor3D<M, N, O, H> as HasDevice>::Device

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

type Reduced = Tensor3D<N, O, P, H>

type DeviceR = <Tensor4D<M, N, O, P, H> as HasDevice>::Device

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

type Reduced = Tensor3D<M, O, P, H>

type DeviceR = <Tensor4D<M, N, O, P, H> as HasDevice>::Device

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

type Reduced = Tensor3D<M, N, P, H>

type DeviceR = <Tensor4D<M, N, O, P, H> as HasDevice>::Device

impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> Reduce1<-1> for Tensor4D<M, N, O, P, H>

type Reduced = Tensor3D<M, N, O, H>

type DeviceR = <Tensor4D<M, N, O, P, H> as HasDevice>::Device