calculate_gain

dragon.vm.torch.nn.init.calculate_gain(
  nonlinearity,
  param=None
)[source]

Return the gain value according to nonlinearity function.

Parameters:
  • nonlinearity (str) – The nonlinearity to compute gain value.
  • param (number, optional) – The optional param value for nonlinearity.
Returns:

number – The gain value.