如何在Python中并行化这个嵌套的for循环？

Question

我有一个numpy 2D数组，我需要在其所有元素上调用一个函数。

我现在的做法是：

h, w, c = x.shape
for i in range( 0, h ):
    for j in range( 0, w ):
        my_func( x[i][j][0], x[i][j][1], x[i][j][2] )

my_func()是一个函数，用于确定x像素的最近颜色，其R，G和B值为x[i][j][0]，x[i][j][1]和{{1}在任何位置x[i][j][2]。

我还在学习Python中的并行编程。在数组[i,j]的不同元素上并行调用此函数的推荐方法是什么？

我也在探索在Tensorflow中将其包装起来的方法，以便能够在GPU上运行它。