剪辑

实现一个clip模块,因为pytorch只有一个简单的clamp函数。

Clip

裁剪张量。

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)

裁剪输入张量。