如何使用GPU在MATLAB中指示3D数组的特定切片？

Question

我在MATLAB中有一个4x4x1250矩阵。我想找到一种方法来逐片移动4x4矩阵，以便分别找到4x4矩阵的条件。

我不想在循环中执行它，因为我想在GPU上执行此操作并希望将其编入索引。

我看到“挤压”，但我认为它不适用于3D阵列...

我有点想使用arrayfun，但我不知道如何指出我感兴趣的特定维度。

有什么想法吗？

编辑：我认为我提供的详细信息已经足够了，但是：

我有一个矩阵A，大小为4x4x1250。

我对构成A的1250 4x4矩阵的条件感兴趣。所以我想说B = A（：，：，1）。

我想计算cond（B），但实际上我想要1250这些计算。

如果我执行arrayfun，我不知道如何指定要沿其切片的A的特定维度。

Answer 1

ARRAYFUN忽略输入的形状，并以纯粹的元素方式运作。 GPU上还有PAGEFUN在数组的页面上运行 - 但是，如果您使用明确支持的功能之一，PAGEFUN只能提供优势 - 否则它以元素方式运行。