transpose method

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

Transpose: swap two dimensions.

Implementation

Tensor transpose(int dim0, int dim1) {
  final order = List<int>.generate(ndim, (i) => i);
  order[dim0] = dim1;
  order[dim1] = dim0;
  return permute(order);
}