pytorch上"变量(data.type(张量)).to(设备)"的含义是什么



有人能向我解释一下下面这行代码的含义是real_skeleton = Variable(data.type(Tensor)).to(device)吗?谢谢

  • Variable已弃用PyTorch函数,可以忽略
  • .type(Tensor)使用张量默认数据类型强制转换data
  • .to(device)device(即CPU、GPU等(发送张量存储

最新更新