Pytorch
关注
分享
扫一扫
hozhangel
这个作者很懒，什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
pytorch 常见函数理解
gather >>> a = torch.Tensor([[1,2],[3,4]]) >>> a tensor([[ 1., 2.], [ 3., 4.]]) >>> torch.gather(a,1,torch.LongTensor([ ... [0,0], ... [1,0]])) tensor([[ 1., 1....原创 2018-11-28 09:40:00 · 422 阅读 · 0 评论 -
查看Ubuntu/cuda/Tensorflow/Pytorch版本
查看Cuda版本： cat /usr/local/cuda/version.txt CUDA Version 9.0.176 查看Linux/Ubuntu版本： （1） cat /proc/version 输出: Linux version 4.4.0-137-generic (buildd@lgw01-amd64-037) (gcc version 5.4.0 2...原创 2018-11-13 09:19:00 · 710 阅读 · 0 评论