Trait dfdx::tensor_ops::Reduce1
source · [−]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>;
}
Expand description
Reduce the I
th dimension of a Tensor. Enables functions like sum_axis() that
reduce values along a single dimension.
Required Associated Types
The resulting tensor type.
The I
th dimension of this can be broadcast into Self via Broadcast1.