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.youkuaiyun.com/manong_wxd/article/details/78590754?utm_source=blogxgwz0
https://blog.youkuaiyun.com/u014134138/article/details/81275139?utm_source=blogxgwz2
参考代码:https://pytorch.org/tutorials/beginner/pytorch_with_examples.html

本文对比了PyTorch与NumPy在常见运算操作上的差异,如矩阵乘法、最大值计算、平方等,并提供了转换代码示例,帮助读者理解两者之间的联系与区别。
1万+

被折叠的 条评论
为什么被折叠?



