PyTorch 速查

img = torch.from_numpy(img).float().to(device)
原文地址:https://www.cnblogs.com/yaos/p/14014185.html