transpose function

Tensor transpose(
  1. Tensor a,
  2. int dim0,
  3. int dim1
)

Implementation

Tensor transpose(Tensor a, int dim0, int dim1) {
  return a.transpose(dim0, dim1);
}