To use nn.Linear module, you have to import torch as below. import torch. 2 Inputs and 1 output (1 neuron). net = torch.nn.Linear(2,1);.
確定! 回上一頁