pub struct AllAxes;
Expand description
Represents all available axes on a tensor.
Trait Implementations
sourceimpl<const M: usize, const N: usize, H: Tape> BroadcastTo<Tensor2D<M, N, H>, AllAxes> for Tensor0D<H>
impl<const M: usize, const N: usize, H: Tape> BroadcastTo<Tensor2D<M, N, H>, AllAxes> for Tensor0D<H>
sourceimpl<const M: usize, const N: usize, const O: usize, H: Tape> BroadcastTo<Tensor3D<M, N, O, H>, AllAxes> for Tensor0D<H>
impl<const M: usize, const N: usize, const O: usize, H: Tape> BroadcastTo<Tensor3D<M, N, O, H>, AllAxes> for Tensor0D<H>
sourceimpl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> BroadcastTo<Tensor4D<M, N, O, P, H>, AllAxes> for Tensor0D<H>
impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> BroadcastTo<Tensor4D<M, N, O, P, H>, AllAxes> for Tensor0D<H>
sourceimpl<T: CountElements, const M: usize> DeviceReduce<[T; M], AllAxes> for Cpuwhere
T::Dtype: CountElements<Dtype = T::Dtype>,
Self: DeviceReduce<T, AllAxes> + FillElements<[T; M]> + FillElements<T::Dtype>,
impl<T: CountElements, const M: usize> DeviceReduce<[T; M], AllAxes> for Cpuwhere
T::Dtype: CountElements<Dtype = T::Dtype>,
Self: DeviceReduce<T, AllAxes> + FillElements<[T; M]> + FillElements<T::Dtype>,
type Reduced = <Cpu as DeviceReduce<T, AllAxes>>::Reduced
type Reduced = <Cpu as DeviceReduce<T, AllAxes>>::Reduced
The smaller type.
sourcefn reduce_into_no_reset<A: Accumulator<T::Dtype>>(
r: &mut Self::Reduced,
t: &[T; M]
)
fn reduce_into_no_reset<A: Accumulator<T::Dtype>>(
r: &mut Self::Reduced,
t: &[T; M]
)
Reduces
T
into Self::Reduced
with accumulator A
without resetting the values in r
.sourcefn broadcast_into_no_reset<A: Accumulator<T::Dtype>>(
t: &mut [T; M],
r: &Self::Reduced
)
fn broadcast_into_no_reset<A: Accumulator<T::Dtype>>(
t: &mut [T; M],
r: &Self::Reduced
)
sourcefn reduce_into<A: Accumulator<T::Dtype>>(r: &mut Self::Reduced, t: &T)
fn reduce_into<A: Accumulator<T::Dtype>>(r: &mut Self::Reduced, t: &T)
Fills
r
with Accumulator::INIT before reducing.sourcefn broadcast_into<A: Accumulator<T::Dtype>>(t: &mut T, r: &Self::Reduced)
fn broadcast_into<A: Accumulator<T::Dtype>>(t: &mut T, r: &Self::Reduced)
Fills
t
with Accumulator::INIT before broadcasting.sourcefn reduce<A: Accumulator<T::Dtype>>(t: &T) -> Box<Self::Reduced>
fn reduce<A: Accumulator<T::Dtype>>(t: &T) -> Box<Self::Reduced>
Allocates
Self::Reduced
then calls DeviceReduce::reduce_into()sourceimpl DeviceReduce<f32, AllAxes> for Cpu
impl DeviceReduce<f32, AllAxes> for Cpu
sourcefn reduce_into_no_reset<A: Accumulator<f32>>(r: &mut Self::Reduced, t: &f32)
fn reduce_into_no_reset<A: Accumulator<f32>>(r: &mut Self::Reduced, t: &f32)
Reduces
T
into Self::Reduced
with accumulator A
without resetting the values in r
.sourcefn broadcast_into_no_reset<A: Accumulator<f32>>(t: &mut f32, r: &Self::Reduced)
fn broadcast_into_no_reset<A: Accumulator<f32>>(t: &mut f32, r: &Self::Reduced)
sourcefn reduce_into<A: Accumulator<T::Dtype>>(r: &mut Self::Reduced, t: &T)
fn reduce_into<A: Accumulator<T::Dtype>>(r: &mut Self::Reduced, t: &T)
Fills
r
with Accumulator::INIT before reducing.sourcefn broadcast_into<A: Accumulator<T::Dtype>>(t: &mut T, r: &Self::Reduced)
fn broadcast_into<A: Accumulator<T::Dtype>>(t: &mut T, r: &Self::Reduced)
Fills
t
with Accumulator::INIT before broadcasting.sourcefn reduce<A: Accumulator<T::Dtype>>(t: &T) -> Box<Self::Reduced>
fn reduce<A: Accumulator<T::Dtype>>(t: &T) -> Box<Self::Reduced>
Allocates
Self::Reduced
then calls DeviceReduce::reduce_into()sourceimpl<T: CountElements> HasAxes<AllAxes> for T
impl<T: CountElements> HasAxes<AllAxes> for T
sourceimpl<const M: usize, const N: usize, const O: usize, H: Tape> Reduce<AllAxes> for Tensor3D<M, N, O, H>
impl<const M: usize, const N: usize, const O: usize, H: Tape> Reduce<AllAxes> for Tensor3D<M, N, O, H>
sourceimpl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> Reduce<AllAxes> for Tensor4D<M, N, O, P, H>
impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> Reduce<AllAxes> for Tensor4D<M, N, O, P, H>
impl<H: Tape> ReduceTo<Tensor0D<H>, AllAxes> for Tensor0D<H>
impl<const M: usize, H: Tape> ReduceTo<Tensor0D<H>, AllAxes> for Tensor1D<M, H>
impl<const M: usize, const N: usize, H: Tape> ReduceTo<Tensor0D<H>, AllAxes> for Tensor2D<M, N, H>
impl<const M: usize, const N: usize, const O: usize, H: Tape> ReduceTo<Tensor0D<H>, AllAxes> for Tensor3D<M, N, O, H>
impl<const M: usize, const N: usize, const O: usize, const P: usize, H: Tape> ReduceTo<Tensor0D<H>, AllAxes> for Tensor4D<M, N, O, P, H>
Auto Trait Implementations
impl RefUnwindSafe for AllAxes
impl Send for AllAxes
impl Sync for AllAxes
impl Unpin for AllAxes
impl UnwindSafe for AllAxes
Blanket Implementations
sourceimpl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
const: unstable · sourcefn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more