Torch::full two overloads

C++
1

Hello all,

in datasets.cpp in pytorch_cpp-master/utils is the line (473)

torch::Tensor class_id = torch::full({}, (long int)this->class_ids.at(idx), torch::TensorOptions().dtype(torch::kLong));

I am not able to compile this with VS2019.

Neither

at::Tensor torch::full(at::IntArrayRef,const c10::Scalar &,c10::TensorOptions)

nor

at::Tensor torch::full(at::IntArrayRef,const c10::Scalar &,c10::optional<at::DimnameList>,c10::TensorOptions)

is hit. What is the correct cast at which argument?

Many thanks for your help.

2 
auto options = torch::TensorOptions().dtype(torch::kLong);
int val = (long int)this->class_ids.at(idx);

torch::Tensor class_id = torch::full({}, val, options);

Does the job.