Torch.squeeze 和 torch.unsqueeze 等效于手电筒(阵列火）

torch.squeeze and torch.unsqueeze equivalent in Flashlight (arrayfire)

本文关键字：手电筒阵列 squeeze torch unsqueeze Torch 更新时间：2023-10-16

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

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

energies = energies.squeeze(-1)

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

您可以使用 af：：moddims 函数执行此操作：

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