pytorch:.cuda() & .cpu() & .data & .numpy()

下面将将tensor转成numpy的几种情况

1. GPU中的Variable变量:

a.cuda().data.cpu().numpy()

2. GPU中的tensor变量:

a.cuda().cpu().numpy()

3. CPU中的Variable变量:
a.data.numpy()

4. CPU中的tensor变量:

a.numpy()

总结:

.cuda()是读取GPU中的数据

.data是读取Variable中的tensor

.cpu是把数据转移到cpu上

.numpy()把tensor变成numpy

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。