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

最新推荐文章于 2025-06-20 14:05:21 发布

原创最新推荐文章于 2025-06-20 14:05:21 发布 · 1.8k 阅读

0 ·

CC 4.0 BY-SA版权

Python 同时被 3 个专栏收录

15 篇文章

订阅专栏

PyTorch

8 篇文章

订阅专栏

编程开发

8 篇文章

订阅专栏

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

部署运行你感兴趣的模型镜像

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]])

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