Simple Artificial Neural Network

1

Hello ,
I’m new to PyTorch, I want to construct a simple Neural Network with one hidden layer (Relu), the input will be a 1930 * 240 tensor, output 1930 * 10. Is there any template code with commentary to learn and thank you.

2

Would this work as a starter code:

nb_samples = 1930
in_features = 240
hidden = 50
out_dim = 10

model = nn.Sequential(
    nn.Linear(in_features, hidden),
    nn.ReLU(),
    nn.Linear(hidden, hidden),
    nn.ReLU(),
    nn.Linear(hidden, out_dim)
)

x = torch.randn(nb_samples, in_features)
output = model(x)
3

Thank you , an easy and a quik example.