pub fn permute<R: Runtime>(
input: &RudaTensor<R>,
axes: &[usize],
) -> Result<RudaTensor<R>>Expand description
Materialize a permutation in a new packed row-major tensor; do not return an aliased view.
pub fn permute<R: Runtime>(
input: &RudaTensor<R>,
axes: &[usize],
) -> Result<RudaTensor<R>>Materialize a permutation in a new packed row-major tensor; do not return an aliased view.