nelement()方法

 

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

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