We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
torch.nn.Linear
torch.nn.Embedding
torch.nn.Conv2d
transformers.pytorch_utils.Conv1D