nelement()方法

 pytorch中的 nelement() 可以统计 tensor 的元素个数。