trainy = trainy.type(torch.FloatTensor)
阅读背景：
torch 写模型遇到 RuntimeError: Found dtype Double but expected Float
来源：互联网
分享到：
