PyTorch 速查 

img = torch.from_numpy(img).float().to(device)
posted @ 2020-01-05 18:41  2021年的顺遂平安君  阅读(155)  评论(0)  编辑  收藏  举报