Skip to main content

permute

ruda_tensor_device::dispatch

Function permute 

Source 
pub fn permute<R>(tensor: RudaTensor<R>, axes: &[usize]) -> RudaTensor<R>
where
    R: Runtime,
Expand description

Permute a tensor’s dimensions