剪辑
实现一个clip模块,因为pytorch只有一个简单的clamp函数。
类
裁剪张量。 |
- class Clip
基础:
Module裁剪张量。
- Parameters:
clip_value_min – 用于裁剪的下限值,可以是数字或张量
clip_value_max – 用于剪裁的上界数值或张量
learn_min – 一个布尔值。如果为True,则学习最小值。clip_value_min将用于初始化。默认值为False
learn_max – 一个布尔值。类似于 learn_min,但用于最大值。
- Raises:
ValueError –
- __init__(clip_value_min, clip_value_max, learn_min=False, learn_max=False)
初始化。
- forward(inputs)
裁剪输入张量。