python - Tensorflow (Python) : How to append a scalar to each row in a tensor

Question

如果我有一个第一个维度是动态的二维张量，我如何在每一行的末尾附加一个标量值？

因此，如果我将 [[1,2], [3,4]] 提供给张量，我想使其成为 [[1,2,5], [3,4,5]]。

示例(不起作用):

a = tf.placeholder(tf.int32, shape=[None, 2])
b = tf.concat([tf.constant(5), a], axis=1)

这给了我:
ValueError:无法连接具有输入形状的“concat_3”(操作:“ConcatV2”)的标量(改用 tf.stack):[]、[?,2]、[]。

我认为这需要 tf.stack、tf.tile 和 tf.shape 的某种组合，但我似乎无法做到正确。

Answer 1

IMO 填充是这里最简单的方法:

a = tf.placeholder(tf.int32, shape=[None, 2])
b = tf.pad(a, [[0, 0], [0, 1]], constant_values=5)

将附加一个 5 -列。
文档: tf.pad 。