pub struct ReduceToDescriptor<const N: usize> {
pub op: ReduceToOp,
pub input_shape: [i32; N],
pub output_shape: [i32; N],
pub element: ElementKind,
}Expand description
Descriptor for a broadcast-reverse reduction.
input_shape is the source extents; output_shape is the target
extents — same rank, with every reduced dim collapsed to size 1
(the caller left-pads output_shape with 1s if the forward
broadcast added leading dims). Per-dim constraint:
output_shape[d] == 1 || output_shape[d] == input_shape[d].
Fields§
§op: ReduceToOpWhich reduction to apply over each output cell’s broadcast set.
input_shape: [i32; N]Input tensor shape.
output_shape: [i32; N]Output tensor shape — input_shape with the reduced dims
collapsed to 1.
element: ElementKindElement type.
Trait Implementations§
Source§impl<const N: usize> Clone for ReduceToDescriptor<N>
impl<const N: usize> Clone for ReduceToDescriptor<N>
Source§fn clone(&self) -> ReduceToDescriptor<N>
fn clone(&self) -> ReduceToDescriptor<N>
Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreimpl<const N: usize> Copy for ReduceToDescriptor<N>
Auto Trait Implementations§
impl<const N: usize> Freeze for ReduceToDescriptor<N>
impl<const N: usize> RefUnwindSafe for ReduceToDescriptor<N>
impl<const N: usize> Send for ReduceToDescriptor<N>
impl<const N: usize> Sync for ReduceToDescriptor<N>
impl<const N: usize> Unpin for ReduceToDescriptor<N>
impl<const N: usize> UnsafeUnpin for ReduceToDescriptor<N>
impl<const N: usize> UnwindSafe for ReduceToDescriptor<N>
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more