Python tf.sparse.transpose用法及代码示例

转置SparseTensor

用法

tf.sparse.transpose(
    sp_input, perm=None, name=None
)

返回的张量维度 i 将对应于输入维度 perm[i] 。如果没有给出perm，则设置为(n-1...0)，其中n是输入张量的秩。因此，默认情况下，此操作对二维输入张量执行常规矩阵转置。

例如，如果 sp_input 具有形状 [4, 5] 和 indices /values ：

[0, 3]:b
[0, 1]:a
[3, 1]:d
[2, 0]:c

那么输出将是形状为 [5, 4] 和 indices /values 的 SparseTensor ：

[0, 2]:c
[1, 0]:a
[1, 3]:d
[3, 0]:b