Pls scroll towards 1/3 down from this page 4
for
equals = top_class == labels
should it be
equals = top_class == labels.view(-1,64)
a small test that I did

1043×936

rather than
equals = top_class == labels.view(*top_class.shape)
*I understood the unpacking of top_class.shape is based on this 12
because the former has a shape of (64,64) based on the explanation of

equals will have shape (64, 64), try it yourself. What it’s doing is comparing the one element in each row of top_class with each element in labels which returns 64 True/False boolean values for each row.This is because of broadcasting.

Pls advise.
Cheers and Happy New Year.