Pytorch
xintao_dashen
这个作者很懒,什么都没留下…
展开
-
pytorch中文文档
https://pytorch-cn.readthedocs.io/zh/latest/torchvision/torchvision-transform/ https://ptorch.com/docs/1/torchlists转载 2018-12-21 22:45:48 · 154 阅读 · 0 评论 -
Pytorch文档——Tensor Attributes -torch.device
torch.device class torch.device A一个torch.device 是一个对象,这个对象代表着torch.tensor在哪个device上,或者会被分配到哪个device。 torch.device 对象包含 device type (‘cpu’ or ‘cuda’) 还有一个可选择的设备类型序列号. 如果设备序列号没有被声明,这代表着使用当前的设备 。例如一个 to...原创 2018-12-07 14:57:57 · 5600 阅读 · 0 评论