AttributeEmbedding

AttributeEmbedding#

class nfflr.nn.AttributeEmbedding(atom_features: Literal['cgcnn', 'basic', 'cfid'] = 'cgcnn', d_model: int = 64)[source]#

Methods

forward(x)