.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/
转载文章受原作者版权保护。转载请注明原作者出处!