torch.numel()简要介绍

定义

torch.numel(input)

用法

返回元素个数

示例

import torch

a = torch.tensor([[1,2,3],[4,5,6]])

torch.numel(a)

>>>6