首页 > 解决方案 > 如何初始化 at::Scalar?

问题描述

我试图调用一个名为linspace的 PyTorch C++ API 函数。当我尝试像 Python 风格那样编写它时,我写道:

torch::Tensor x = torch::linspace(-1, 1, 100);

这段代码出现了分段错误:

[1]    6038 segmentation fault  ./example-app

我在 pytorch.org 中阅读了 C++ 文档,它给出了 linspace 的格式:

at::Tensor torch::linspace(const at::Scalar &start, const at::Scalar &end, c10::optional<int64_t> steps = c10::nullopt, at::TensorOptions options = {})

在这里我不知道at::Scalar到底是什么意思,所以我在关于at namespace 的文档中搜索了它,但我没有找到任何名为Scalar的类。

但是,我确实在at命名空间中发现了许多具有标量类型参数的函数。

这是关于linspace函数的文档,这是关于at namespace 的文档。

标签: c++apipytorchlibtorch

解决方案


推荐阅读