As part of an assignment I constructed a simple MLP with 2 hidden, 1 output layer into a Sequential()
In one of the tests, it gets 1D vector [4] as input, but instead of returning [3] length output, it returns [4, 3]
What on earth causes this? The same construct returns 1D tensors for 1D tensors in earlier tests.
This is on pytorch.

self.a_net = nn.Sequential(
nn.Linear(n_node_features*2,96),
nn.ReLU(),
nn.Linear(96,96),
nn.ReLU(),
nn.Linear(96,n_edge_features)
)