How do you create a linear transformation in nn.Module, with 784 inputs and 256 outputs?
a) nn.Linear(784, 256)
b) nn.Linear(256, 784)
c) nn.Linear(784, 784)
d) nn.Linear(256, 256)