modelzoo.transformers.pytorch.layers_api_demo.pytorch_transformer.TransformerModel#
- class modelzoo.transformers.pytorch.layers_api_demo.pytorch_transformer.TransformerModel[source]#
Bases:
torch.nn.Module
Methods
- param src
Tensor, shape [seq_len, batch_size]
init_weights
- __call__(*args: Any, **kwargs: Any) Any #
Call self as a function.
- __init__(ntoken: int, d_model: int, nhead: int, d_hid: int, nlayers: int, dropout: float = 0.5, activation: str = 'gelu', max_len: int = 10)[source]#
- static __new__(cls, *args: Any, **kwargs: Any) Any #