pytorch笔记 torch.clamp(截取上下限)

Posted UQI-LIUWJ

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了pytorch笔记 torch.clamp(截取上下限)相关的知识,希望对你有一定的参考价值。

1 基本用法

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

使得tensor中比min小的变成min,比max大的变成max

2 使用举例

import torch
a = torch.randn(4)
print(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])


以上是关于pytorch笔记 torch.clamp(截取上下限)的主要内容,如果未能解决你的问题,请参考以下文章

PyTorch中的torch.clamp()实现矩阵裁剪

Pytorch Note33 数据增强

PyTorch笔记

pytorch笔记:policy gradient

比用Pytorch框架快200倍!0.76秒后,笔记本上的CNN就搞定了MNIST | 开源

李宏毅机器学习 p5(pytorch 1)学习笔记