Python调用第三方库Numpy实现对两个矩阵的卷积

钟会 发布于 29 天前 102 次阅读


前言-关于Nmupy的安装

我们可以在在终端或命令提示符中运行以下命令安装NumPy

pip3 install numpy

默认情况使用国外线路,国外太慢,我们可以使用清华的镜像:

pip3 install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple

安装完成后,可以通过以下代码令验证 NumPy 是否安装成功:

import numpy as np
print(np.__version__)

如果安装成功,会输出正确的版本号信息。

正文

话不多说,代码如下:

import numpy as np
# 20 * 20
a = np.random.randint(0,9,(20, 20))

#  3 * 3
b = np.random.randint(0,9,(3, 3))

# 输出行数 = 输入行数 - 卷积核行数 + 1
# 输出列数 = 输入列数 - 卷积核列数 + 1
h = len(a) - len(b) + 1
l = len(a[0]) - len(b[0]) + 1

# 输出矩阵
c = np.zeros((h, l), dtype = int)

for i in range(h):
    for j in range(l):
        put = a[i:i + 3, j:j + 3]
        c[i][j] = np.sum(put * b)

print("矩阵:")
print(a)
print('\n')
print("卷积核:")
print(b)
print('\n')
print("结果:")
print(c)

希望对你有所帮助。

好好爱自己
最后更新于 2025-11-04