ontolearn.nces_modules

NCES modules.

Classes

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

Module Contents

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

Bases: torch.nn.Module

MAB module.

dim_V

num_heads

fc_q

fc_k

fc_v

fc_o

forward(Q, K)

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

Bases: torch.nn.Module

SAB module.

mab

forward(X)

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

Bases: torch.nn.Module

ISAB module.

I

mab0

mab1

forward(X)

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

Bases: torch.nn.Module

PMA module.

S

mab

forward(X)