neural_tangents.stax.Sin

neural_tangents.stax.Sin(a=1.0, b=1.0, c=0.0)[source]

Affine transform of Sin nonlinearity, i.e. a sin(b*x + c).

Parameters:
  • a (float) – output scale.

  • b (float) – input scale.

  • c (float) – input phase shift.

Return type:

tuple[InitFn, ApplyFn, LayerKernelFn]

Returns:

(init_fn, apply_fn, kernel_fn).