pytorch号称神经网络界的numpy,确实非常好用,容易上手,自己在学习中总结了 pytorch 与numpy 的一些运算操作上的不同.
np.dot(a,b) ==> torch.mm(a,b)
np.maxmum(h,0) ==> h.clamp(min=0)
np.square(x) ==> x.pow(2)
y=x.copy() ==> y=x.clone()
np.random.randn() ==> torch.randn()
转置: x.T ==> x.t()
更多可参考:https://blog.csdn.net/manong_wxd/article/details/78590754?utm_source=blogxgwz0
https://blog.csdn.net/u014134138/article/details/81275139?utm_source=blogxgwz2
参考代码:https://pytorch.org/tutorials/beginner/pytorch_with_examples.html