Referring to https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/core/Tensor.h 7, I could not find an API equivalent to tensor.nelement() to get total number of elements in the tensor regardless of its dimension. Does anyone know about it?