美文网首页
tensorflow中Tensor与numpy.ndarray的

tensorflow中Tensor与numpy.ndarray的

作者: LoveToday2020 | 来源:发表于2020-04-16 09:46 被阅读0次

1.Tensor转换成numpy.ndarray

c = tf.constant([[1,2,4]])
d = tf.constant([[3,4,5]])
x = concatenate([c,d], axis=1)
sess = tf.InteractiveSession()
# x.eval()等价于sess.run(x)
y = x.eval()
y, type(y)

此时得到的结果是

(array([[1, 2, 4, 3, 4, 5]], dtype=int32), numpy.ndarray)

如果

x = concatenate([c,d], axis=0)

得到的结果则是

(array([[1, 2, 4],
        [3, 4, 5]], dtype=int32), numpy.ndarray)

2.numpy.ndarray转换成tf.Tensor

w = np.ndarray([2,3])
z = tf.convert_to_tensor(w)
z, type(z)

得到的结果是

(<tf.Tensor 'Const_20:0' shape=(2, 3) dtype=float64>,
 tensorflow.python.framework.ops.Tensor)

相关文章

网友评论

      本文标题:tensorflow中Tensor与numpy.ndarray的

      本文链接:https://www.haomeiwen.com/subject/lmplvhtx.html