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(截取上下限)的主要内容,如果未能解决你的问题,请参考以下文章