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