Java语言实现矩阵卷积运算

B站2020校招笔试真题 ----- 通过100%

注意：same卷积，没有填充0！！

此程序要求输入两个矩阵，矩阵A任意形状即可，只要大于矩阵B即可，

输入要求：

第一行： 输入两个整数，代表矩阵A的行数和列数

接下来输入n行，即输入矩阵A

接下来输入矩阵B（卷积核的高）

接下输入m行

Eg: 3 3

      40 24 135

      200 239 238

      90 34 94

      2

      0.0   0.6

      0.1    0.3

最后输出运算结果，也是一个矩阵。

代码如下：

import java.util.Scanner;

public class test {
    public static void main(String[] args) {
        //输入
        Scanner sc = new Scanner(System.in);
        int h = sc.nextInt();
        int w = sc.nextInt();
        int[][] matrix = new int[h][w];
        for (int i = 0; i < h; i++) {
            for (int j = 0; j < w; j++) {
                matrix[i][j] = sc.nextInt();
            }
        }
        int m = sc.nextInt();
        float[][] kernel = new float[m][m];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < m; j++) {
                kernel[i][j] = sc.nextFloat();
            }
        }
        //计算
        int[][] res = new int[h - m + 1][w - m + 1];
        for (int i = 0; i < h - m + 1; i++) {
            for (int j = 0; j < w - m + 1; j++) {
                res[i][j] = matmul(matrix, kernel, i, j, m);
            }
        }
        for(int i = 0; i < h - m + 1; i++){
            for(int j = 0; j < w - m + 1; j++){
                System.out.print(res[i][j] + " ");
            }
            System.out.println();
        }

    }

    public static int matmul(int[][] a, float[][] b, int l1, int l2, int len) {
        float res = 0;
        for (int i = l1; i < l1 + len; i++) {
            for (int j = l2; j < l2 + len; j++) {
                res += a[i][j] * b[i - l1][j - l2];
            }
        }
        return (int)res;
    }
}