transpose¶
根据 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])