tf.sparse.transpose

Транспонирует SparseTensor

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

Размерность i возвращаемого тензора будет соответствовать размерности входного тензора perm[i]. Если perm не указано, оно устанавливается в (n-1...0), где n — ранг входного тензора. Таким образом, по умолчанию данная операция выполняет обычную транспозицию матрицы для входных тензоров 2-го порядка.

Например, если sp_input имеет форму [4, 5] и indices / values:

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

то выход будет SparseTensor формы [5, 4] и indices / values:

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