transpose

paddle.sparse. transpose ( x, perm, name=None ) [源代码]

根据 perm 对输入的 x 维度进行重排,但不改变数据, x 必须是多维 SparseTensor 或 COO 格式的 2 维或 3 维 SparseTensor。

\[out = transpose(x, perm)\]

参数

  • x (Tensor) - 输入的 Tensor,数据类型为 float32、float64、int32 或 int64。

  • perm (list|tuple) - perm 长度必须和 x 的维度相同,并依照 perm 中数据进行重排。

  • name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

转置后的稀疏 Tensor, 数据类型和压缩格式与 x 相同。

代码示例

import paddle

dense_x = paddle.to_tensor([[-2., 0.], [1., 2.]])
sparse_x = dense_x.to_sparse_coo(1)
out = paddle.sparse.transpose(sparse_x, [1, 0])