博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
tensor与数组转化
阅读量:6828 次
发布时间:2019-06-26

本文共 686 字,大约阅读时间需要 2 分钟。

import tensorflow as tf

img1 = tf.constant(value=[[[[1],[2],[3],[4]],[[1],[2],[3],[4]],[[1],[2],[3],[4]],[[1],[2],[3],[4]]]],dtype=tf.float32)
img2 = tf.constant(value=[[[[1],[1],[1],[1]],[[1],[1],[1],[1]],[[1],[1],[1],[1]],[[1],[1],[1],[1]]]],dtype=tf.float32)
img = tf.concat(values=[img1,img2],axis=3)      #<class 'tensorflow.python.framework.ops.Tensor'>
sess=tf.Session()

#sess.run(tf.initialize_all_variables())

sess.run(tf.global_variables_initializer())

#转化为numpy数组

img_numpy=img.eval(session=sess) #<class 'numpy.ndarray'>

#转化为tensor

img_tensor= tf.convert_to_tensor(img_numpy) #<class 'tensorflow.python.framework.ops.Tensor'>

转载于:https://www.cnblogs.com/hapyygril/p/10945633.html

你可能感兴趣的文章