torch.clamp()函数

最新推荐文章于 2025-07-20 00:18:56 发布

原创最新推荐文章于 2025-07-20 00:18:56 发布 · 4.3w 阅读

69 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#python

Python 专栏收录该内容

25 篇文章

订阅专栏

本文介绍了如何使用torch.clamp()函数将Tensor中元素限制在指定区间，通过实例展示了如何按列操作，确保数值在0到448之间。展示了Python在数据分析中的实用操作。

Qwen3-32B-Chat 私有部署镜像 | RTX4090D 24G 显存 CUDA12.4 优化版

Qwen

文本生成

Qwen3

本镜像基于 RTX 4090D 24GB 显存 + CUDA 12.4 + 驱动 550.90.07 深度优化，内置完整运行环境与 Qwen3-32B 模型依赖，开箱即用。

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

将输入input张量每个元素的范围限制到区间 [min,max]，返回结果到一个新张量。

input (Tensor) – 输入张量
min (Number) – 限制范围下限
max (Number) – 限制范围上限
out (Tensor, optional) – 输出张量

栗子：

boxes_nms=torch.randint(-100,1000,(3,4))
print(boxes_nms)

在这里插入图片描述

接下来对tensor按列操作：

boxes_nms[:, 0] = torch.clamp(boxes_nms[:, 0], min=0)
boxes_nms[:, 1] = torch.clamp(boxes_nms[:, 1], min=0)
boxes_nms[:, 2] = torch.clamp(boxes_nms[:, 2], max=448)
boxes_nms[:, 3] = torch.clamp(boxes_nms[:, 3], max=448)
print(boxes_nms)

意思就是按列对原tensor进行操作，限定了每列中元素的大小范围，若是小于0，则替换为0.若是大于448，则替换为448.
在这里插入图片描述
可看到，一二列中的小于0的元素被替换为了0.三四列中大于448的元素被替换为了448！
Python确实在数据分析方面很强大！！！

您可能感兴趣的与本文相关的镜像