將tensorflow Tensor列解包為單獨的變量

Question

鑒於形式的張量

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

如何將T分成兩個獨立的形狀張量（？，1），每列一個？

Answer 1

您可以使用python的切片表示法切出張量。

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

在內部，它委托給Tensor.__getitem__ ，並作為圖的一部分進行評估。