torch.clamp()

本文介绍了PyTorch中的torch.clamp()和clamp_()函数,用于限制张量元素的取值范围。torch.clamp()不改变原始张量,返回新张量;clamp_()则直接在原张量上修改。示例展示了如何使用这两个函数,并对比了它们的效果。理解这些函数对于张量操作和神经网络训练至关重要。
摘要由CSDN通过智能技术生成

1.用法示例

torch.clamp(input, min, max, out=None) → Tensor

将输入input张量每个元素的“夹紧”到区间 [min,max],并返回结果到一个新张量。

当然,需要注意的在于,input张量的dtype和[min,max]的数据类型
保持一致??

数学表示如下:

      | min, if x_i < min
y_i = | x_i, if min <= x_i <= max
      | max, if x_i > max

Example:

a=torch.randint(low=0,high=10,size=(10,1))
print(a)
a=torch.clamp(a,3,9)
print(a)

结果
在这里插入图片描述

torch.clamp()和np.clip()应该是差不多的

2.clamp_下划线

有下划线的表示修改并付给自身,无下划线的表示需要返回处理后的值

h = k.clamp(min=0) #将结果存入h,k保留原值

k.clamp_(min=0)   # 将结果存入k

3.参考

1.pytorch:torch.clamp()
2.pytorch clamp与clamp_区别

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值