Output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28 ...

Output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28 ...

More to explore