本文介绍了NumPy库中的`arange`函数，用于创建等差数列，详细解释了其参数含义，并通过实例展示了不同参数设置下的用法。此外，还探讨了张量操作中的`unsqueeze`和`squeeze`函数，它们分别用于增加和减少张量的维度。`unsqueeze`在指定维度上增加一个维度，而`squeeze`则移除大小为1的维度。这些基础知识对于理解和操作多维数据至关重要。
1.np.arange()
只需记住第一个是起始点，第二个终点，第三个是步长，终点必须有，起始点默认为0，步长默认为1
#一个参数 默认起点0，步长为1
a = np.arange(3)
#两个参数 默认步长为1
a = np.arange(5,8)
#三个参数 起点为0，终点为5，步长为0.1
a = np.arange(0, 5, 0.1)
2.unsqueeze()函数起升维的作用,参数表示在哪个地方加一个维度。对于tensor而言
a.unsqueeze(dim=0)
在dim=0维度加1维
3，相应的，squeeze()函数起降维的作用
扫一扫
2452
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
