pytorch
文章平均质量分 77
纽约恋情
这个作者很懒,什么都没留下…
展开
-
Pytorch中permute(),transpose(),view()函数
比如原向量中(1, 2, 3, 4),1的下标是0,2的下标是1,3的下标是2,4的下标是3;**这个意思不是torch.transpose()只能作用于二维向量,它的意思是一次只能进行两个维度的转置,如果需要多个维度的转置,那么需要多次调用transpose()。2.维度变换函数是进行的浅拷贝操作(只复制了指像某个对象的指针,新旧对象还是共享同一块内存)即view操作会连带原来的变量一同变形,这是不合法的,所以也会报错;而tensor.view()维度变形函数的要求是需要Tensor的内存连续。原创 2023-07-26 11:19:43 · 434 阅读 · 1 评论 -
【torch.nn.init】初始化参数方法解读
稀疏矩阵:将2D输入张量填充为稀疏矩阵,其中非零元素将从正态分布N ( 0 , 0.01 ) N(0,0.01)N(0,0.01)中提取。正态分布:从给定均值和标准差的正态分布N(mean, std)中生成值,填充输入的张量或变量。xavier_normal 分布:用一个正态分布生成值,填充输入的张量或变量。kaiming_normal 分布:用一个正态分布生成值,填充输入的张量或变量。均匀分布: 从均匀分布N(a,b)中生成值,填充输入的张量或变量。正交矩阵:用一个(半)正交矩阵填充输入张量。原创 2023-04-27 11:25:45 · 2406 阅读 · 0 评论 -
python torch.div( ) 函数使用说明
参考链接: torch.div() 二.解释: Divides each element of the input input with the scalar other and returns a new resulting tensor. outi = inputi / other if input is of type FloatTensor or DoubleTensor ,other should be a real number, othe原创 2023-04-24 16:14:40 · 682 阅读 · 0 评论