tf.raw_ops.ReadVariableXlaSplitND

Разделяет тензор входной переменной ресурса по всем измерениям.

tf.raw_ops.ReadVariableXlaSplitND(
    resource, T, N, num_splits, paddings=[], name=None
)

Операция, которая разделяет тензор входной переменной ресурса на основе заданного атрибута num_splits, дополняет срезы необязательно и возвращает срезы. Срезы возвращаются в порядке «строка-столбец».

Эта операция может быть сгенерирована через мост TPU.

Например, для тензора input:

[[0, 1, 2],
 [3, 4, 5],
 [6, 7, 8]]

num_splits:

[2, 2]

и paddings:

[1, 1]

ожидаемый результат outputs:

[[0, 1],
 [3, 4]]
[[2, 0],
 [5, 0]]
[[6, 7],
 [0, 0]]
[[8, 0],
 [0, 0]]