I tried to extend the pixel averaging MNIST classifier from lesson 4 to test more digits, starting with just adding 4s to the 3s and 7s in the lesson. However when trying to use:

on valid_3_tens - the stacked tensor of all the 3s I get the error:

Can I still use broadcasting for this?

Hi Danie
Try return (mnist_distance(x,mean3) < mnist_distance(x,mean7) ) and ( … )
Regards Conwyn