You can specify a group of axis dimensions (axes) to transpose the input tensor data. If no axis dimension is specified, the tensor data is transposed in reverse order by default.
This function supports only tensor data on the device.
transpose_operator(input: Tensor, axes: List[int])
Parameter |
Type |
Description |
|---|---|---|
input |
Tensor class |
Tensor class to be transposed. 2-, 3-, and 4-dimension are supported, and the data type can be base.dtype.float32, base.dtype.float16, and base.dtype.uint8. For details, see Tensor Class Attributes. |
axes |
List[int] |
Transposition option. The length must be the same as the dimension of the input tensor.
|
Transposed tensor data