Size([1, 28, 28]) Note that the reshaped tensor's shape has to have the same number of total elements as the original. If you try flat_tensor.reshape(3,28 ...
確定! 回上一頁