Rust Arrayfire - 使用一个数组索引到另一个数组

问：

在过去的三天里，我一直在尝试在我的机器学习项目中实现损失函数（交叉熵），遵循哈里森·金斯利（Harrison Kinsley）的《从头开始的神经网络》一书。

到目前为止，我从softmax激活层得到了一个带有类预测的数组：

[0.3298     0.3351     0.3351]
[0.3298     0.3351     0.3351]
[0.3298     0.3351     0.3351]
[0.3298     0.3351     0.3351]
...

以及带有目标标签的数组： [0, 2, 1, 1 ...]

为了计算损失，我可以取所有正确置信度的负自然对数的平均值。 （-log（0.3298） + -log（0.3351） + -log（0.3351） + -log（0.3351）） / 4

但是我坚持从类预测数组中提取这些“正确的置信度分数”。
有没有办法直接使用目标标签数组来使用 arrayfire rust 绑定来做到这一点，例如 numpy 可以做到：

predictions[range(l), target_labels]

我尝试过使用 Seqs 向量，但如果样本量增加，它似乎效率低下。

对此的任何帮助将不胜感激。 （如果这个问题是重复的，那么我就傻傻地找到原来的帖子）