I am considering the sample code from the documentation:
import torch from torch import nn # m = nn.Linear(20, 30) input = torch.randn(128, 20) output = m(inp