Im new to PyTorch and I keep getting the error mat1 dim1 must match mat1 dim0
mat1 dim1 must match mat1 dim0
this is my code for the network
class Net(Module): def _