Pytorch
积累实验、程序常用到的方法等
ZcsLv_zyq
这个作者很懒,什么都没留下…
展开
-
Pytorch之张量维度及不同维度的操作
参考: pytorch 矩阵维度 - 搜索结果 - 知乎 Pytorch 中的 dim操作介绍 - 大数据 - 亿速云 1.如何理解dim? pytorch的dim和numpy的axis很类似 不同dim的数据长什么样? 维度为0, 0维张量也叫标量 1 维度为1, 0维张量也叫矢量 [1,2] 维度为2, 0维张量也叫原创 2021-12-15 16:53:38 · 3668 阅读 · 1 评论 -
pytorch之expand repeat
1. expand tensor.expand(*sizes) 1 expand函数用于将张量中单数维的数据扩展到指定的size。 首先解释下什么叫单数维(singleton dimensions),张量在某个维度上的size为1,则称为单数维。比如zeros(2,3,4)不存在单数维,而zeros(2,1,4)在第二个维度(即维度1)上为单数维。expand函数仅仅能作用于这些单数维的维度上。 参数*sizes用于逐个指定各个维度扩展后的大小(也可以理解为拓展的次数),对于不需要或者无法(即非单数维)原创 2021-12-12 15:19:31 · 958 阅读 · 0 评论 -
pytorch之cat()
在给定维度上对输入的张量序列seq进行连接操作 >>> x = torch.randn(2, 3) >>> x 0.5983 -0.0341 2.4918 1.5981 -0.5265 -0.8735 [torch.FloatTensor of size 2x3] #0表示按列,1表示按行 >>> torch.cat((x, x, x), 0) 0.5983 -0.0341 2.4918 1.5981 -0.5265 -0.873.原创 2021-12-12 14:11:56 · 634 阅读 · 0 评论 -
Pytorch之view()方法
原创 2021-12-12 14:08:13 · 608 阅读 · 0 评论