将tensorflow Tensor列解包为单独的变量

时间:2018-06-10 06:52:28

标签: python tensorflow tensor

给出形式的张量

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

如何将T分成两个独立的形状张量(?,1),每列一个?

1 个答案:

答案 0 :(得分:1)

您可以使用python的切片表示法来切割张量。

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

在内部,此委托给Tensor.__getitem__,并作为图表的一部分进行评估。