e3j.utils.options.LinearIndexwiseInitialization

class e3j.utils.options.LinearIndexwiseInitialization

LinearIndexwise weight initialization options.

Values:
  • FAN_IN – stddev is 1/sqrt(m_in) per block (e3nn: “path”).

  • FAN_OUT – stddev is 1/sqrt(m_out) per block (e3nn: “irrep”).

  • FAN_IN_FCTP – stddev is 1/sqrt(m_in * num_indices), matching the effective normalization of a FullyConnectedTensorProduct with a one-hot species input.

Methods

parse(key)

parse_value(key)

__getitem__(name)

Return the member matching name.

__iter__()

Return members in definition order.

__len__()

Return the number of members (no aliases)

Attributes

FAN_IN

FAN_OUT

FAN_IN_FCTP