tf2.x 中tensor numpy 和list的转换

原创已于 2022-03-17 10:17:42 修改 · 1.8k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#list #tensorflow #深度学习

于 2021-12-17 18:26:15 首次发布

tensorflow2.x 专栏收录该内容

39 篇文章

订阅专栏

本文介绍了在TensorFlow中如何将Tensor转换为NumPy数组及列表，并演示了如何设置Tensor变量的值。主要内容包括：使用`.numpy()`方法将Tensor转换为NumPy数组，将NumPy数组进一步转换为Python列表，以及通过`K.set_value()`函数更新Tensor变量。

部署运行你感兴趣的模型镜像

#tensor 常量 转 numpy
a = tf.constant([1,2,3])
a_numpy = a.numpy()

#tensor 转list
a = tf.constant([1,2,3])
a_numpy = list(a.numpy())

tensor变量转numpy

#tensor 常量 转 numpy
a = tf.Variable([1,2,3])
a_numpy = list(a.numpy())

tensor 变量设置值

import tensorflow as tf
import tensorflow.python.keras.backend as K 

varib = tf.Variable([1,2])
new_value = np.array([2,3])
K.set_value(varib, new_value)
print(K.eval(varib))
print(varib.numpy())

您可能感兴趣的与本文相关的镜像