当前位置:   article > 正文

警告解决方案:To copy construct from a tensor, it is recommended to use..._data.tensordatasetto copy construct from a tensor,

data.tensordatasetto copy construct from a tensor, it is recommended to use

代码

x_train, y_train, x_valid, y_valid = map(
    torch.tensor, (x_train, y_train, x_valid, y_valid)
)
  • 1
  • 2
  • 3

在这里插入图片描述

警告:

D:\Users\ThinkStation\anaconda3\envs\pytorch\lib\site-packages\ipykernel_launcher.py:2: UserWarning: To copy construct from a tensor, it is recommended to use sourceTensor.clone().detach() or sourceTensor.clone().detach().requires_grad_(True), rather than torch.tensor(sourceTensor).
  • 1

解决方法:
torch.tensor改成torch.as_tensor

x_train, y_train, x_valid, y_valid = map(
    torch.as_tensor, (x_train, y_train, x_valid, y_valid)
)
  • 1
  • 2
  • 3
本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号