【Tensorflow】将Eager Tensors转化为numpy数组
·
将Eager Tensors转化为numy数组
某些时候需要将numpy数组转化为tensor类型进行操作,这时候会使用 tf.convert_to_tensor()进行转化:
import tensorflow as tf
import numpy as np
numpyData = np.random.random(5)
tensorData = tf.convert_to_tensor(numpyData, dtype=tf.float32)
此时tensorData变为了Eager Tensors类型,但是当我们想要通过经常使用的tensorData.eval()操作将其转化为numpy数组时,这是会报错,对某些需要来说这就会很尴尬。
解决方法:
使用.numpy()函数,或者也可以执行numpy.array(tensorData )
numpyOut = numpyData .numpy()
或:
numpyOut = numpy.array(tensorData )
AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。
更多推荐


所有评论(0)