Question

我有一个形状为NxM的Tesnor。

我想创建另一个具有相同形状的Tensor，一直填充到一个特定的列（每行可能不同），其余部分填充另一个值（例如10）。

我该怎么做？谢谢。

Answer 1

类似的事情可以为您提供帮助：

input = tf.Variable([[1, 2, 3, 4, 5], [1, 2, 3, 4, 5], [1, 2, 3, 4, 5]], dtype = tf.float32)
indices = tf.constant([1, 4, 2])
X = tf.ones_like(input)
Y = tf.constant(10, dtype = tf.float32, shape = input.shape)
result = tf.where(tf.sequence_mask(indices, tf.shape(input)[1]), X, Y)
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print(sess.run(input))
    print(sess.run(indices))
    print(sess.run(result))

Tensorflow：如何创建具有特定值的Tensor？

1 个答案: