把tensor转为numpy_NumPy和Tensor之间得转换

本文介绍了如何在TensorFlow的tf.Tensors和NumPy的ndarray之间进行转换。TensorFlow操作会自动将NumPy数组转化为Tensors,而NumPy操作则会自动将Tensors转化为NumPy数组。通过.tensorm()方法可以显式地将Tensor转化为numpy数组,但此过程可能涉及GPU到主机内存的复制。
摘要由CSDN通过智能技术生成

Converting between a TensorFlow tf.Tensors and a NumPy ndarray is easy:

1.TensorFlow operations automatically convert NumPy ndarrays to Tensors.

2,NumPy operations automatically convert Tensors to NumPy ndarrays.

Tensors are explicitly converted to NumPy ndarrays using their .numpy() method. These conversions are typically cheap since the array and tf.Tensor share the underlying memory representation, if possible. However, sharing the underlying representation isn’t always possible since the tf.Tensor may be hosted in GPU memory while NumPy arrays are always backed by host memory, and the conversion involves a copy from GPU to host memory.

import numpy as np

ndarray = np.ones([3, 3])

print("TensorFlow operations convert numpy arrays to Tensors automatically")

tensor = tf.multiply(ndarray, 42)

print(tensor)

print("And NumPy operations convert Tensors to numpy arrays automatically")

print(np.add(tensor, 1))

print("The .numpy() method explicitly converts a Tensor to a numpy array")

print(tensor.numpy())

标签:operations,转换,Tensor,numpy,print,NumPy,Tensors

来源: https://blog.csdn.net/weixin_43824178/article/details/99208554

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值