Machine-Learning-Collection
Machine-Learning-Collection copied to clipboard
change the dimension of x
Dimension of x is [128,1,28,28]. But x dimension should be [128,784] That's why it show :---- RuntimeError: mat1 dim 1 must match mat2 dim 0