AtomicNumberEmbedding# class nfflr.nn.AtomicNumberEmbedding(d_model: int = 64, num_embeddings: int = 118)[source]# Methods forward(x)