transpose

Returns x transposed over axes. If no axes are provided, tensor is transposed over all axes (that is, the order of the axes are reversed).

Note: API is numpy transpose, aka torch permute. https://docs.scipy.org/doc/numpy/reference/generated/numpy.transpose.html

Example:

val t = FloatTensor(Shape(1, 2, 3), floatArrayOf(1f, 2f, 3f, 4f, 5f, 6f)) t.transpose(0, 2, 1) FloatTensor(Shape(1, 3, 2), floatArrayOf(1f, 3f, 5f, 2f, 4f, 6f))
Content copied to clipboard
Content copied to clipboard