rml_op_transpose_params
Parameters for the RML_OP_TRANSPOSE operation.
Transposes the input
tensor similar to numpy.transpose
. For example, when perm=(1, 0, 2), given an input tensor of shape (1, 2, 3), the output shape will be (2, 1, 3).
Note
This operator currently is ONLY for internal usage. Don’t use it directly.
API
Graph Manipulation
Items
Name |
Type |
Description |
---|---|---|
input |
Operation that outputs tensor containing input data. |
|
num_axes |
size_t |
Number of axes that MUST be equal to rank of input tensor. |
axes[RML_TENSOR_MAX_RANK] |
const int32_t* |
A nullable list of axis. |