PyTorch
行路南
公众号:CV前沿
展开
-
【PyTorch】标量、向量、张量的直观理解
标量标量就是一个数字。标量也称为0维数组。比如5套房子中的“5”就是标量;向量向量是一组标量组成的列表。向量也称为1维数组。比如房子的价格是受多种因素(是否为学区房、附近有无地铁、房子面积、房间数量、楼层等)来影响,那么我们将这多种因素来表示为房子的特征,这一组特征值就可以用向量表示。矩阵矩阵是由一组向量组成的集合。矩阵也称为2维数组。在刚才的例子中,一套房子的特征可以用一个向量来表示。那么我们要建m套房子的数据集,那么就是m个向量的组合,也即是得到一个m行n列的矩阵。(n为一套房子的向量的原创 2021-03-25 19:40:09 · 2953 阅读 · 0 评论 -
【PyTorch】torch.mean(), dim=0, dim=1 详解
创建一个tensor,这个tensor是一个元素类型为浮点型的2维数组import torchs = torch.arange(6,dtype=float).reshape((2,3))print(s)print(s.shape)# 查看tensor的形状tensor([[0., 1., 2.], [3., 4., 5.]], dtype=torch.float64)torch.Size([2, 3])dim属性的全称是dimension,表示维度。dim=0为第0个维度原创 2021-03-25 12:20:09 · 20082 阅读 · 5 评论