c++ - 检查 torch::Tensor 是否为空的最佳方法是什么？

Question

我目前正在使用 has_storage()检查张量是否为空的方法，但我想知道除此之外还有什么更好的方法!除了初始化 torch::Tensor 的事实之外，使用它是否还有其他含义。总是有一个存储，而一个空的没有!

Answer 1

经过一番挖掘，发现最好的解决方案是使用 .numel()返回张量具有的元素数的方法。
总之:

要知道是否分配了张量(类型和存储)，请使用 defined()。

要知道分配的张量是否有零个元素，请使用 numel()

要知道是否分配了张量以及它是否有零个元素，请使用 defined()然后 numel()

边注 :
当 torch::Tensor t; 时，空张量(例如使用 .numel() 创建的张量)返回零。用来。而大小/大小将导致异常。
这是对返回空张量(在我上面刚刚解释的意义上)的情况的完美检查，您可以简单地执行以下操作:

if (!tensor.numel())
{
    std::cout<<"tensor is empty!" << std::endl;
    // do other checks you wish to do
}

ref