14、Python 数值计算中的矩阵操作与矩阵乘积

Python 数值计算中的矩阵操作与矩阵乘积

在 Python 中进行数值计算时，矩阵操作是非常重要的一部分。下面将详细介绍矩阵的创建、函数应用、转置与重塑、广播以及矩阵乘积等内容。

1. 矩阵尺寸获取

在 PyTorch 中，可以使用 size() 方法并结合 dim 参数来获取矩阵特定维度的大小。例如：

hv_cnts_pt.size(dim=0) 
# 5 
hv_cnts_pt.size(dim=1) 
# 26 

和其他 Python 索引一样，也可以使用负数， -1 表示最后一个维度：

hv_cnts_pt.size(dim=-1) 
# 26 

2. 矩阵创建函数

NumPy 和 PyTorch 都提供了创建特定初始值的 (m, n) 矩阵的函数，具体如下表所示：
| 初始值情况 | NumPy 函数 | PyTorch 函数 |
| ---- | ---- | ---- |
| 填充零 | np.zeros((m,n)) | torch.zeros((m,n)) |
| 填充一 | np.ones((m,n)) | torch.ones((m,n))