ontolearn.nces_modules

NCES modules.

Module Contents

Classes

MAB	MAB module.
SAB	SAB module.
ISAB	ISAB module.
PMA	PMA module.

class ontolearn.nces_modules.MAB(dim_Q, dim_K, dim_V, num_heads, ln=False)

Bases: torch.nn.Module

MAB module.

forward(Q, K)

class ontolearn.nces_modules.SAB(dim_in, dim_out, num_heads, ln=False)

Bases: torch.nn.Module

SAB module.

forward(X)

class ontolearn.nces_modules.ISAB(dim_in, dim_out, num_heads, num_inds, ln=False)

Bases: torch.nn.Module

ISAB module.

forward(X)

class ontolearn.nces_modules.PMA(dim, num_heads, num_seeds, ln=False)

Bases: torch.nn.Module

PMA module.

forward(X)