我有一个3D ndarray ( X )表示的卷,它的值在0和255之间,还有另一个3D ndarray ( Y ),它是第一个数组的任意掩码,值为0或1。我在R方面的经验如下:也许R的优点是我可以对3D数组进行线性索引,因为在numpy中只使用一个索引就可以得到一个2D矩阵。我想它可能涉及到numpy.random.choice,但似乎我不能有条件地使用它
我有一个函数,它以numpy数组(A)作为输入。根据数学计算,这个数组可以是一个2d的数组,也可以是一个三维的数组。有一个整数m,它可以是任意数,除非数组是2D,否则m的值总是0。因为A既可以是3D的,也可以是2D的,所以我尝试了3D风格的切片。of m is 0 if A is 2DIndexError: too many indices for array
如果A是2D的话,我想把