PyTorch
关注
分享
扫一扫
Microstrong0305
菜鸟程序猿
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
PyTorch中torch.cat
import torcha = torch.ones([1,2])print(a)# tensor([[1., 1.]])b = torch.ones([1,2])print(b)# tensor([[1., 1.]])print(torch.cat([a,b],1))# tensor([[1., 1., 1., 1.]])torch.cat就是将a,b 按行放在一起，...原创 2018-08-27 14:43:20 · 1388 阅读 · 0 评论 -
PyTorch使用多GPU
在PyTorch中使用多GPU则需要在声明模型之后，对声明的模型进行初始化，如：model = DataParallel(modelName)然后，直接运行模型训练的Python文件之后，则默认使用所有的GPU。相关文章：【1】Pytorch使用多GPU【2】pytorch DataParallel 多GPU使用...原创 2019-04-13 17:06:40 · 1078 阅读 · 0 评论