Java语言实现矩阵卷积运算

谁借莪１个温暖的怀抱￠ 2021-10-13 02:02 458阅读 0赞

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

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

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

**输入要求**：

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

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

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

接下输入m行

**Eg**: 3 3

40 24 135

200 239 238

90 34 94

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;
        }
    }