shape函数

在NumPy中，shape函数用于获取数组的形状，即返回一个元组，表示数组每个维度的大小。例如，对于一个二维数组a，a.shape将返回一个包含两个整数的元组，分别表示数组的行数和列数。具体使用方法如下：

import numpy as np

# 创建一个二维数组
a = np.array([[1, 2, 3], [4, 5, 6]])

# 输出数组的形状
print(a.shape)  # (2, 3)

这里输出的元组`(2, 3)`表示数组`a`有两行三列。在机器学习和深度学习中，经常需要用到shape函数来确定输入数据的形状，以便进行模型的搭建和参数的初始化。