pytorch clamp()使用

最新推荐文章于 2024-01-05 21:14:07 发布

不知道叫啥好一点

最新推荐文章于 2024-01-05 21:14:07 发布

阅读量1.4k

点赞数 1

分类专栏： pytorch入坑指南文章标签： Pytorch clamp

本文链接：https://blog.csdn.net/A_A666/article/details/108754443

版权

pytorch入坑指南专栏收录该内容

14 篇文章 6 订阅

订阅专栏

参考：https://pytorch.org/docs/stable/generated/torch.clamp.html#torch.clamp

使用格式

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

在最大值和最小值之间对input进行裁剪
$\mathrm{y}_{\mathrm{i}}=\left\{\begin{array}{ll} \min & \text { if } \mathrm{x}_{\mathrm{i}}<\min \\ \mathrm{x}_{\mathrm{i}} & \text { if } \mathrm{min} \leq \mathrm{x}_{\mathrm{i}} \leq \max \\ \max & \text { if } \mathrm{x}_{\mathrm{i}}>\max \end{array}\right.$

Pytorch代码

>>> a = torch.randn(4)
>>> a
tensor([-1.7120,  0.1734, -0.0478, -0.0922])
>>> torch.clamp(a, min=-0.5, max=0.5)
tensor([-0.5000,  0.1734, -0.0478, -0.0922])

另一种使用方式(代码来自faster_rcnn)

cls_bbox[..., 0] = (cls_bbox[..., 0]).clamp(min=0, max=width)
cls_bbox[..., 2] = (cls_bbox[..., 2]).clamp(min=0, max=width)
cls_bbox[..., 1] = (cls_bbox[..., 1]).clamp(min=0, max=height)
cls_bbox[..., 3] = (cls_bbox[..., 3]).clamp(min=0, max=height)