pytorch 与numpy 部分操作的对应关系

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