nelement()方法

 

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

posted @ 2020-05-21 09:15  像阳光，像春天  阅读(1190)  评论(0)    收藏  举报