pytorch统计模型参数量可以使用param.numel()来实现,接下来的这篇文章我们就来看看到底怎么实现吧。
在pytorch中有一个numel函数。通过这个函数我们可以得知tensor中一共包含多少个元素,接下来这篇文章我们就来了解一下numel函数的用法说明吧。