LINEAR nn.linear in_feature out_feature

CLASS torch.nn.Linear(in_feature, out_feature, bias)

m = nn.Linear(20,30)
input = torch.randn(128,20)
output = m(input)
print(output.size()) # torch.Size([128,30])

 

ref. https://pytorch.org/docs/stable/generated/torch.nn.Linear.html?highlight=nn+linear#torch.nn.Linear