将 Pytorch 代码转换为 C++ torch 代码 (Even =total_real[:, :, :, :, 0])

Question

以下 pytorch 代码运行良好。我想将其转换为 C++ torch 代码。

even = total_real[:, :, :, :, 0] # extract even elements
odd = total_real[:, :, :, :, 1] # extract odd elements

total_real.shape 是 火炬.Size([1, 1, 1023, 64, 2])

偶数形状是 火炬.Size([1, 1, 1023, 64])

奇怪的形状是 火炬.Size([1, 1, 1023, 64])

在 C++ Torch 库中， 如何将其转换为相同？

在 C++ 火炬代码中，

even = total_real[":", ":", ":", ":", 0]; # extract even elements
odd = total_real[":", ":", ":", ":", 1]; # extract odd elements

even.shape 是 [ CPUFloatType{1,1023,64,2} ]，而不是 [1, 1, 1023, 64]。

Answer 1

找到的解决方案如下。

自动偶=total_real.index({"...", 0}); 自动奇数=total_real.index({"...", 1});

致以诚挚的问候，