如何在C ++中遍历Pytorch张量中的每个值？

Question

我正在尝试为C ++中的PyTorch张量做类似[0，如果我<1其他1]。 我尝试使用tensor.accessor（），但似乎它要求您事先知道尺寸。我想动态传递它。

反正我可以在C ++中为Pytorch做到这一点吗？

Answer 1

看看是否有帮助：

使用TensorIterator：

https://labs.quansight.org/blog/2020/04/pytorch-tensoriterator-internals/

或利用t.is_contiguous（）/ t.contiguous（）简化横向：

https://discuss.pytorch.org/t/iterating-over-tensor-in-c/60333/2