Struct SortBackwardArgs 

pub struct SortBackwardArgs<'a, T: Element> {
    pub dy: TensorRef<'a, T, 2>,
    pub indices: TensorRef<'a, i32, 2>,
    pub dx: TensorMut<'a, T, 2>,
}

Args bundle for a sort_backward launch.

Fields

dy: TensorRef<'a, T, 2>

Upstream grad of sorted-values output [batch, row_len].

indices: TensorRef<'a, i32, 2>

Saved indices from FW [batch, row_len].

dx: TensorMut<'a, T, 2>

Grad of the original input [batch, row_len] (output).