transpose_shape

axonml_tensor::shape

Function transpose_shape 

Source 
pub fn transpose_shape(
    shape: &[usize],
    dim0: usize,
    dim1: usize,
) -> Result<Shape>
Expand description

Computes the shape after transposing dimensions.

§Arguments

  • shape - Input shape
  • dim0 - First dimension
  • dim1 - Second dimension

§Returns

Transposed shape and strides modifier.