transpose
- Graph.transpose(a, perm=None, *, name=None)
Reorder the dimensions of a tensor.
- Parameters
a (np.ndarray or Tensor) – The tensor whose dimensions you want to permute, \(x\).
perm (list[int] or np.ndarray(int), optional) – The order of the input dimensions for the returned tensor. If you provide it, it must be a permutation of all integers between 0 and
N-1
, where N is the rank of a. If you don’t provide it, the order of the dimensions is inverted, that is to say, it defaults to[N-1, …, 1, 0]
.name (str, optional) – The name of the node.
- Returns
The input tensor with its dimensions permuted. The i-th dimension of the returned tensor corresponds to the perm[i]-th input dimension.
- Return type