Struct RepeatBackwardArgs 

pub struct RepeatBackwardArgs<'a, T: Element, const N: usize> {
    pub dy: TensorRef<'a, T, N>,
    pub dx: TensorMut<'a, T, N>,
}

Args bundle for a Repeat backward launch.

dy.shape must match the forward output shape (input_shape[d] * repeats[d] per axis). dx.shape must match desc.input_shape. No saved forward tensors are needed — the BW formula is a pure sum over dy.

Fields

dy: TensorRef<'a, T, N>

Upstream gradient — full forward output shape.

dx: TensorMut<'a, T, N>

Gradient w.r.t. the forward input — input shape.