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.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 2.9

PyTorch 2.9

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值