permute

permute(
    x: &Tensor<T>,
    axes: 
        &[i64]
        | &[i64; _]
        | [i64; _] 
        | Vec<i64> 
        | &Vec<i64>
) -> Result<Tensor<T>, TensorError>

use hpt::{
    common::TensorInfo,
    error::TensorError,
    ops::{ShapeManipulate, TensorCreator},
    Tensor,
};
fn main() -> Result<(), TensorError> {
    // Create a tensor with shape [2, 3, 4]
    let a = Tensor::<f32>::zeros(&[2, 3, 4])?;

    // Permute dimensions to [4, 2, 3]
    let b = a.permute(&[2, 0, 1])?;
    println!("{}", b.shape());

    // Permute dimensions to [3, 4, 2]
    let c = a.permute(&[1, 2, 0])?;
    println!("{}", c.shape());

    // This will return an error as [1, 1, 0] is not a valid permutation
    let d = a.permute(&[1, 1, 0]);
    assert!(d.is_err());

    Ok(())
}