pytorch近似运算

.floor() 往下取整数

.ceil() 网上取整数

.round() 四舍五入 >=0.5向上取整,<0.5向下取整

.trunc() 裁剪,只取整数部分

.frac() 只取小数部分

 

如何查看梯度的模

w.grad.norm(2) 梯度的二范数,可以用于观测梯度是否爆炸(一般都比较小,10左右,100都算比较大了)

梯度裁剪时,会用到,gradient clipping

a = torch.rand(2, 3)*15
print(a)
#clamp(min), 将小于min的值变为min
b = a.clamp(10)
print(b)

#out: tensor([[ 2.3666,  5.2836,  4.2964],
#out:         [ 8.5975, 11.7687,  5.1429]])
#out: tensor([[10.0000, 10.0000, 10.0000],
#out:         [10.0000, 11.7687, 10.0000]])

#另一种是.clamp(min,  max)
#将小于min的变为min,将大于max的变为max

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值