在Python中创建不带for循环的内核矩阵

import numpy as np

# 创建一个3x3的内核矩阵
kernel = np.array([[1, 2, 1],
                   [2, 4, 2],
                   [1, 2, 1]])

# 创建一个5x5的输入矩阵
input_matrix = np.array([[1, 2, 3, 4, 5],
                         [6, 7, 8, 9, 10],
                         [11, 12, 13, 14, 15],
                         [16, 17, 18, 19, 20],
                         [21, 22, 23, 24, 25]])

# 使用NumPy的卷积函数进行内核矩阵与输入矩阵的卷积运算
output_matrix = np.convolve(input_matrix.flatten(), kernel.flatten(), mode='same').reshape(input_matrix.shape)

print(output_matrix)