将 tensorflow 张量列解压缩为单独的变量

Question

给定一个形式的张量

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

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

Answer 1

您可以使用 python 的 slice notation 对张量进行切片。

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

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