首页 文章

如何将numpy数组的张量转换为张量?

提问于
浏览
1

我有一个numpy数组列表,如下所示:

a=np.array([tf.convert_to_tensor(1),tf.convert_to_tensor(2)])

我想将此列表转换为张量 . 我的真实列表不像常量示例,而是一些复杂的张量,所以有谁知道如何做到这一点?

1 回答

  • 1

    我假设所有张量都具有相同的形状 . 然后你可以调用tf.stack

    >>> print(tf.stack([tf.convert_to_tensor(1), tf.convert_to_tensor(2)]))
    Tensor("stack:0", shape=(2,), dtype=int32)
    

    请注意,它接受列表,而不是numpy数组 .

相关问题