美文网首页2022
pytorch:.cuda() & .cpu() & .data

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

作者: 不太聪明的亚子 | 来源:发表于2020-08-16 17:05 被阅读0次

    下面将将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

        本文链接:https://www.haomeiwen.com/subject/biatjktx.html