Solved by ptrblck in post #2 
              
              
                Your last error message might be a misleading error due to a missing batch dimension in your input. 
It has been fixed in master. 
Could you unsqueeze at dim0 and try it again?