Solved by ptrblck in post #4 
              
              
                Ah ok, broadcasting was introduced in version 0.2. 
Try to update Pytorch to version 0.3. 
Just select your config and follow the instructions here.