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

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

  1. GPU中的Variable变量：

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

  1. GPU中的tensor变量：

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

  1. CPU中的Variable变量：
    a.data.numpy()

  2. CPU中的tensor变量：

a.numpy()

总结：

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

.data是读取Variable中的tensor

.cpu是把数据转移到cpu上

.numpy()把tensor变成numpy