LibTorch sizeof张量

Question

我想知道如何获取 torch::Tensor.
数据类型的字节大小 我在网上某个地方看到 sizeof(tensor.dtype()) 应该可以工作，但是对于我的 float32 张量，它打印出 1.

Answer 1

终于自己找到了
torch::elementSize() returns 给定 ScalarType 的大小。

要将类型为 caffe2::MetaType 的 tensor.dtype() 转换为标量，我必须使用此转换器 torch::typeMetaToScalarType(tensor.dtype())

因此，一个张量在内存中的大小可以这样计算：
tensor.numel() * torch::elementSize(torch::typeMetaToScalarType(tensor.dtype()))

文档提到了 none 这些函数，因此很难找到它们。