您想摆脱那些
for循环。尝试这个:
def exampleKernelA(M, x, N, y): """Example kernel function A""" i, j = np.indices((N, M)) # Define the custom kernel function here kernel[i, j] = np.sqrt((x[i, 0] - y[j, 0]) ** 2 + (x[i, 1] - y[j, 1]) ** 2) return kernel
您也可以通过广播来做到这一点,它可能更快,但是来自的直观性却有所下降
MATLAB。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)