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

你可能感兴趣的:(pytorch:.cuda() & .cpu() & .data & .numpy())