Struct Permutation 

pub struct Permutation;

Tensor permutation (axis shuffle + optional unary op): B[mB] = α * op_a(A[mA]).

Implementations

impl Permutation

pub unsafe fn new<'h>( handle: &'h Handle, a: &TensorDescriptor<'h>, modes_a: &[i32], op_a: UnaryOp, b: &TensorDescriptor<'h>, modes_b: &[i32], compute_desc: *const c_void, ) -> Result<OperationDescriptor<'h>>

Safety

compute_desc must be null or valid.

Trait Implementations

impl Debug for Permutation

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.