swapaxes function

VARP swapaxes(
  1. VARP a,
  2. int axis1,
  3. int axis2
)

Implementation

VARP swapaxes(VARP a, int axis1, int axis2) {
  final ndim = a.ndim;
  MnnAssert(ndim != null, "a must have a shape");
  final axes = List.generate(ndim!, (index) => index);
  final tmp = axes[axis1];
  axes[axis1] = axes[axis2];
  axes[axis2] = tmp;
  return F.transpose(a, axes);
}