pytorch 学习–60分钟入个门

简单相加
a+b

torch.add(a,b)
给出一个输出向量
torch.add(a,b,out=result)

原地相加
b.add_(a) # 这时值已经给了 b

print(y[:,1])

a = torch.rand(5,3)
a.numpy()

b = torch.from_numpy(a)

展开阅读全文

上一篇：Keras使用多个GPU并行下一篇：Windows下Caffe在GPU编译过程