在Python中,`shape`是一个属性,通常用于获取数组或矩阵的维度信息。`shape`属性返回一个元组,其中每个元素代表对应维度的长度。对于一维数组,`shape`返回一个整数,表示数组的长度;对于多维数组或矩阵,`shape`返回一个元组,表示各个维度的长度。
例如,在使用NumPy库创建一个二维数组时,可以通过`arr.shape`获取数组的形状:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr.shape) 输出:(2, 3)
在PyTorch中,`shape`属性用于获取张量的维度信息,其中`shape[i]`表示张量的第`i`维的大小,`shape[-1]`表示张量最后一维的大小。
需要注意的是,`shape`属性是特定于NumPy和PyTorch等库的,并不是Python语言本身的一部分。在不同的库中,`shape`的使用方法和含义可能略有不同。