如何在 LibTorch (C++) 中将 torch::Torch 的类型从 Float 更改为 Long

Question

我正在使用 LibTorch (PyTorch C++ API) 在 C++ 中编码。在这里，我传递 predicted_value 和 target_value，它们都是 torch::Tensor，大小为 {1, 1}。

torch::Tensor loss = torch::nll_loss(predicted_value, target_value);

当我尝试评估以上内容时，出现以下错误：

 0.4997 [ Variable[CPUFloatType]{1,1} ]   # printout of predicted_value
-0.5392 [ Variable[CPUFloatType]{1,1} ]   # printout of target_value
terminate called after throwing an instance of 'c10::Error'
  what():  Expected object of scalar type Long but got scalar type Float for argument #2 'target' in call to _thnn_nll_loss_forward (checked_dense_tensor_unwrap at ../../aten/src/ATen/Utils.h:84)

我尝试搜索如何将浮点型张量转换为长型张量，但只能找到 Python 的文档。非常感谢解决此问题的建议！

Answer 1

tensor.to(torch::kLong) 给你 Long 类型。

这里是 Tensor 的 to 函数的重载定义：

inline Tensor Tensor::to(ScalarType dtype, bool non_blocking, bool copy) const {
    static auto table = globalATenDispatch().getOpTable("aten::to(Tensor self, ScalarType dtype, bool non_blocking=False, bool copy=False) -> Tensor");
    return table->getOp<Tensor (const Tensor &, ScalarType, bool, bool)>(tensorTypeIdToBackend(type_id()), is_variable())(*this, dtype, non_blocking, copy);
}

Answer 2

inline Tensor Tensor::to(ScalarType dtype, bool non_blocking, bool copy) const {
    static auto table = globalATenDispatch().getOpTable("aten::to(Tensor self, ScalarType dtype, bool non_blocking=False, bool copy=False) -> Tensor");
    return table->getOp<Tensor (const Tensor &, ScalarType, bool, bool)>(tensorTypeIdToBackend(type_id()), is_variable())(*this, dtype, non_blocking, copy);
}

如何在 LibTorch (C++) 中将 torch::Torch 的类型从 Float 更改为 Long

How to change type of torch::Torch from Float to Long in LibTorch (C++)

c++

libtorch