PyTorch 速查

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