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

最新推荐文章于 2025-05-28 10:42:57 发布

danerli

最新推荐文章于 2025-05-28 10:42:57 发布

阅读量1.8k

点赞数

CC 4.0 BY-SA版权

分类专栏： Python PyTorch 编程开发

本文链接：https://blog.csdn.net/danerer/article/details/82908459

Python 同时被 3 个专栏收录

15 篇文章

订阅专栏

PyTorch

8 篇文章

订阅专栏

编程开发

8 篇文章

订阅专栏

本文详细介绍了PyTorch中的clamp函数，该函数用于将输入张量的每个元素限制在一个指定范围内。通过实例演示了如何使用clamp函数，并展示了其在处理随机生成的4x1张量时的效果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

torch.clamp(input, min, max, out=None) → Tensor
将输入input张量每个元素的夹紧到区间 [min,max]，并返回结果到一个新张量。
$\begin{cases} min, & \text{if $input $ <= min} \\ input, & \text{if min<$input $ <max}\\ max, & \text{if $input$ >= max} \end{cases}$

import torch
x = torch.randn(4, 1)

print('tensor原型:',x)

print(torch.clamp(x, -0.5, 0.5))

运行结果：

tensor原型: tensor([[-0.6572],
        [ 0.8355],
        [-0.9475],
        [-0.0340]])
tensor([[-0.5000],
        [ 0.5000],
        [-0.5000],
        [-0.0340]])