用tf将数组转化为tensor

可以使用 TensorFlow 的 tf.convert_to_tensor() 函数将 NumPy 数组转换为 TensorFlow 张量。以下是一个示例代码：

import tensorflow as tf
import numpy as np

# 创建一个 NumPy 数组
arr = np.array([1, 2, 3, 4, 5])

# 将 NumPy 数组转换为 TensorFlow 张量
tensor = tf.convert_to_tensor(arr)

# 打印结果
print(tensor)

输出结果：

tf.Tensor([1 2 3 4 5], shape=(5,), dtype=int64)

注意，tf.convert_to_tensor() 函数默认将 NumPy 数组转换为 TensorFlow 张量的 dtype 为 float32。如需指定 dtype，可以使用 dtype 参数。例如：

import tensorflow as tf
import numpy as np

# 创建一个 NumPy 数组
arr = np.array([1, 2, 3, 4, 5])

# 将 NumPy 数组转换为 TensorFlow 张量，指定 dtype 为 int32
tensor = tf.convert_to_tensor(arr, dtype=tf.int32)

# 打印结果
print(tensor)

输出结果：

tf.Tensor([1 2 3 4 5], shape=(5,), dtype=int32)