class documentation

class Transformer(nn.Module): (source)

Constructor: Transformer(width, layers, heads, attn_mask)

View In Hierarchy

Undocumented

Method __init__ Undocumented
Method forward Undocumented
Instance Variable layers Undocumented
Instance Variable resblocks Undocumented
Instance Variable width Undocumented
def __init__(self, width: int, layers: int, heads: int, attn_mask: torch.Tensor = None): (source)

Undocumented

def forward(self, x: torch.Tensor): (source)

Undocumented

Undocumented

resblocks = (source)

Undocumented

Undocumented