mindchemistry.e3.nn

API Name

Description

Supported Platforms

mindchemistry.e3.nn.Activation

Activation function for scalar-tensors.

Ascend

mindchemistry.e3.nn.BatchNorm

Batch normalization for orthonormal representations.

Ascend

mindchemistry.e3.nn.FullyConnectedNet

Fully-connected Neural Network with normalized activation on scalars.

Ascend

mindchemistry.e3.nn.Gate

Gate activation function.

Ascend

mindchemistry.e3.nn.NormActivation

Activation function for the norm of irreps.

Ascend

mindchemistry.e3.nn.soft_one_hot_linspace

Projection on a basis of functions.

Ascend

mindchemistry.e3.nn.soft_unit_step

Smooth version of the unit step function.

Ascend

mindchemistry.e3.nn.SoftOneHotLinspace

Projection on a basis of functions.

Ascend