Pythonの組み込み関数(42)、numel()

numel()関数:配列の要素数を返します。

例:

params = sum(p.numel() for p in list(net.parameters())) / 1e6 # numel()

print('#Params: %.1fM' % (params))

net.parameters():Pytorchの使用法、ネットネットワークのパラメーターを返すために使用

params:ネットネットワーク内のパラメーターの総数を返すために使用されます

 

943の元の記事を公開 136のような 330,000以上を訪問

おすすめ

転載: blog.csdn.net/weixin_36670529/article/details/105188314