TensorFlow-numpy转换为tensor

最新推荐文章于 2024-11-09 14:57:22 发布

原创最新推荐文章于 2024-11-09 14:57:22 发布 · 1.9k 阅读

5 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

数字图像处理

当前文章被收录于：

数字图像处理

42 篇文章 3 人学习

数字图像处理的内容

订阅专栏查看详情

当前文章被以下社区和专栏收录：

本文演示了如何使用Python的NumPy库创建一维数组，并将其转换为TensorFlow张量。通过具体代码实例，展示了从NumPy数组到TensorFlow张量的转换过程及其输出结果。

import tensorflow as tf
import numpy as np
a = np.arange(0,5)
b = tf.convert_to_tensor(a,dtype=tf.int64)  #将numpy数据格式转换为tensor格式
print("a:",a)
print("b:",b)

"""
a: [0 1 2 3 4]
b: tf.Tensor([0 1 2 3 4], shape=(5,), dtype=int64)
"""