Распаковать тензорные столбцы Tensor в отдельные переменные

Question

Учитывая тензор формы

T = tf.placeholder(tf.float32, [None, 2])

Как я могу разделить T на два отдельных тензора формы (?, 1), по одному на каждый столбец?

Answer 1

Вы можете вырезать тензоры, используя нотацию Python slice .

x = T[:, 0, None]
y = T[:, 1, None]

Внутренне это делегирует Tensor.__getitem__ и оценивается как часть графа.