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