Broadcasting

Part 1 (2020)
1

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:

Screenshot 2021-06-21 at 14.37.49
Screenshot 2021-06-21 at 14.37.492154×36 11.9 KB

on valid_3_tens - the stacked tensor of all the 3s I get the error:
Screenshot 2021-06-21 at 14.39.48
Screenshot 2021-06-21 at 14.39.481296×52 10.3 KB

Can I still use broadcasting for this?

2

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