1. 程式人生 > >pytorch 與numpy 部分操作的對應關係

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