multidimensional-array - 选择任意维度进行过滤？

Question

在 Julia 中，是否有一种“选择在任意维度上循环”的好方法 d？例如，我想将扩散过滤器应用于我想做的 2D x

for j = 1:size(x,2)
  for i = 2:size(x,1)-1
    x2[i,j] = x[i-1,j] - 2x[i,j] + x[i+1,j]
  end
end

但我想写一个函数 diffFilter(x2,x,d) 其中 x 可以是任意维数组，而 d 是小于 ndims(x) 的任何维度，它会沿维度 x[i-1] + 2x[i] - x[i+1] 应用此过滤器 d(进入 x2 没有分配)。知道如何进行索引以便我可以使用 d 使循环的特殊部分成为第 d 索引吗？

Answer 1

您需要查看 Tim Holy 就此主题撰写的两篇博文:

• http://julialang.org/blog/2016/02/iteration
• http://julialang.org/blog/2016/03/arrays-iteration

这应该让你开始了解这个主题。