Issue on docs - flux-2-klein

[rafalromanowicz-sp]

Path: /tutorials/flux/flux-2-klein

Following the instructions results in:

mat1 and mat2 shapes cannot be multiplied (512x2560 and 7680x3072)  

I've tried many changes suggested on various places, however it still doesn't help.