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.
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)
Thank you , an easy and a quik example.