在Tensorflow中，有没有一种从标量张量构造数组张量的直接方法？

Question

例如，类似于：

x = 12
y = np.array([[x, x], [0, 0]])

在Tensorflow中给定标量张量x是否可以直接构造数组张量？不使用tf.expand_dims和tf.pad。

x = tf.placeholder(tf.int32)
y = ?

非常感谢。

Answer 1

您可以使用tf.convert_to_tensor()功能执行此操作：

x = tf.placeholder(tf.int32)  # NOTE: You should probably pass `shape=[]` as well.
y = tf.convert_to_tensor([[x, x], [0, 0]])

另请注意，TensorFlow将隐式调用tf.convert_to_tensor()对任何期望tf.Tensor作为输入的函数的输入，因此您可以将[[x, x], [0, 0]]直接传递给许多函数。 / p>