Function caffe2_math::transpose

source ·

pub fn transpose<TIndex, TData, Context>(
    ndim: i32,
    dims: *const TIndex,
    axes: *const i32,
    x: *const TData,
    y: *mut TData,
    context: *mut Context
)

Expand description

| Transpose tensor X with dims by axes | and write the result to tensor Y. |