Invalid argument 2: size '[-1 x 400]' is invalid

Yeah you can’t view a tensor of that shape as [-1, 400]

1 Like