runtimeerror: output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28] site:stackoverflow.com
By Brainy Butterfly at May 11 2021