深度学习基础系列:矩阵乘法

矩阵乘法

矩阵乘法是矩阵运算的基础,简单来说,对于两个AB矩阵来说,只要A的列数等于B的行数,那么这两个矩阵就可以发生运算

A是一个m x n的矩阵, B是一个n x k的矩阵,那么运算后的结果就是m x k的矩阵

首先把矩阵转换为np.array类型,然后判断它们的类型,查看是否可以相乘

1
2
3
4
5
6
7
8
import numpy as np
def matrix_dot_vector(a:list[list[int|float]],b:list[int|float])-> list[int|float]:
	a = np.array(a)
	b = np.array(b)
	if a.shape[1] != b.shape[0]:
		return -1
	c = np.dot(a, b)
	return c

np.dot()接受两个数组(矩阵)并返回它们的结果

花有重开日,人无再少年
使用 Hugo 构建
主题 StackJimmy 设计