Kyon Huang

AssertionError for the more-than-two-dimensional input of nn.Linear

6

Considering that the input of the linear layer is not necessarily only two-dimensional, I think these changes * change `assert len(inp.size()) == 2 and len(out.size()) == 2` to `assert len(inp.size())...