TensorFlow 张量转换为 NumPy 数组:两种方法详解
在 TensorFlow 中,将数据转换为 NumPy 数组有两种方法:
- 使用 'numpy()' 方法:可以使用 'numpy()' 方法将 TensorFlow 张量转换为 NumPy 数组。例如:
import tensorflow as tf
import numpy as np
# 创建一个 TensorFlow 张量
x = tf.constant([[1, 2], [3, 4]])
# 将 TensorFlow 张量转换为 NumPy 数组
x_np = x.numpy()
print(x_np)
输出:
array([[1, 2],
[3, 4]], dtype=int32)
- 使用 'eval()' 方法:可以使用 'eval()' 方法将 TensorFlow 张量转换为 NumPy 数组。例如:
import tensorflow as tf
import numpy as np
# 创建一个 TensorFlow 张量
x = tf.constant([[1, 2], [3, 4]])
# 将 TensorFlow 张量转换为 NumPy 数组
with tf.Session() as sess:
x_np = x.eval()
print(x_np)
输出:
array([[1, 2],
[3, 4]], dtype=int32)
原文地址: https://www.cveoy.top/t/topic/ncdc 著作权归作者所有。请勿转载和采集!