張量流。 [batch_size, 1] & [batch_size] 的區別

Question

在詞嵌入的 tensorflow 教程中發現：

# Placeholders for inputs
train_inputs = tf.placeholder(tf.int32, shape=[batch_size])
train_labels = tf.placeholder(tf.int32, shape=[batch_size, 1])

這兩個占位符之間可能有什么區別。 它們不是大小為 batch_size 的 int32 列向量嗎？

謝謝。

Answer 1

我通過一些調試找到了答案。

[batch_size] = [ 0, 2, ...]
[batch_size, 1] = [ [0], [2], ...]

雖然仍然不知道為什么使用第二種形式。

Answer 2

train_inputs是行向量，而train_labels是列向量。