xnn.gnn.models.blocks.shifted_softplus

xnn.gnn.models.blocks.shifted_softplus(x)

Shifted softplus log(0.5 e^x + 0.5) (NequIP’s ShiftedSoftPlus).

Parameters:

x (Tensor)

Return type:

Tensor