手电筒中的 torch.squeeze 和 torch.unsqueeze 等效项（arrayfire）答案

【问题标题】：torch.squeeze and torch.unsqueeze equivalent in Flashlight (arrayfire)手电筒中的 torch.squeeze 和 torch.unsqueeze 等效项（arrayfire）
【发布时间】：2019-07-24 02:12:48
【问题描述】：

我正在将 PyTorch 代码移植到 Flashlight 代码中。 Pytorch 中 squeeze 和 unsqueeze 的 Arrayfire 或 Flashlight 函数等效项是什么？

processed_query = self.query_layer(query.unsqueeze(1))

energies = energies.squeeze(-1)

如何将其转换为 Arrayfire 代码？（或者，手电筒？）

【问题讨论】：

标签： c++ flashlight arrayfire

【解决方案1】：

您可以使用af::moddims 函数来做到这一点：

array a = randu(10, 1, 10, 10);
squeezed_a = moddims(a, 10, 10, 10);

【讨论】：