Hardshrink

class paddle.nn. Hardshrink ( threshold=0.5, name=None ) [source]

Hardshrink Activation

\[\begin{split}hardshrink(x)= \\left\\{ \\begin{aligned} &x, & & if \\ x > threshold \\\\ &x, & & if \\ x < -threshold \\\\ &0, & & if \\ others \\end{aligned} \\right.\end{split}\]
Parameters
  • threshold (float, optional) – The value of threshold for hardthrink. Default is 0.5

  • name (str, optional) – Name for the operation (optional, default is None). For more information, please refer to Name.

Shape:
  • input: Tensor with any shape.

  • output: Tensor with the same shape as input.

Examples

import paddle

x = paddle.to_tensor([-1, 0.3, 2.5])
m = paddle.nn.Hardshrink()
out = m(x) # [-1., 0., 2.5]
forward ( x )

Defines the computation performed at every call. Should be overridden by all subclasses.

Parameters
  • *inputs (tuple) – unpacked tuple arguments

  • **kwargs (dict) – unpacked dict arguments

extra_repr ( )

Extra representation of this layer, you can have custom implementation of your own layer.