File size: 204 Bytes
9622166 |
1 2 3 4 5 6 7 8 |
import torch
import torch.nn as nn
#simple test for the linear function to determine it's properties
m = nn.Linear(20, 30)
input = torch.randn(128, 20)
output = m(input)
print(output.size())
print(output) |