PeriodicTableEmbedding

PeriodicTableEmbedding#

class nfflr.nn.PeriodicTableEmbedding(d_model: int = 64)[source]#

Additive row and column embedding.

Methods

forward(zs)

reset_parameters