pytorch 统计tensor中总元素数量的函数：numel()

本文已参与「新人创作礼」活动，一起开启掘金创作之路。

用法示例：

import torch
a = torch.randn(2,3,4)
print(a.numel())    # 24

统计model中所有可训练参数量：

num_params = sum(p.numel() for p in model.parameters())

注：numel() 是pytorch的函数，只适用于 tensor，不能用于统计 list、tuple、dict 等的元素数量。