Pytorch中 .numpy() .item() .cpu() 区别

.numpy() 与 .item()

这两个可以归为一类,是将Tensor变量转换为非Tensor变量。
t.numpy()将Tensor变量转换为ndarray变量,其中t是一个Tensor变量, 可以是标量,也可以是向量,转换后dtype与Tensor的dtype一致。

t.item()将Tensor变量转换为python标量(int float等),其中t是一个Tensor变量, 只能是标量,转换后dtype与Tensor的dtype一致。

.cpu()

将数据的处理设备从其他设备(如.cuda()拿到cpu上),不会改变变量类型,转换后仍然是Tensor变量。

https://blog.csdn.net/u012177700/article/details/106984537

Original: https://blog.csdn.net/weixin_43135178/article/details/117561239
Author: 马鹏森
Title: Pytorch中 .numpy() .item() .cpu() 区别

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/760559/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球