我有一个numpy数组列表,如下所示:
a=np.array([tf.convert_to_tensor(1),tf.convert_to_tensor(2)])
我想将此列表转换为张量。 我的真实列表不像常量示例,而是一些复杂的张量,所以有人知道如何做到这一点吗?
答案 0 :(得分:1)
我认为所有张量都具有相同的形状。然后你可以拨打tf.stack
:
>>> print(tf.stack([tf.convert_to_tensor(1), tf.convert_to_tensor(2)]))
Tensor("stack:0", shape=(2,), dtype=int32)
请注意,它接受列表,而不是numpy数组。