pytorch
文章平均质量分 57
不要清汤锅
The lion doesn't concern himself with the opinions of the sheep
展开
-
PyTorch中的view的用法
在PyTorch中view函数作用为重构张量的维度,相当于numpy中的resize()的功能,但是用法不太一样import torchtt1=torch.tensor([-0.3623,-0.6115,0.7283,0.4699,2.3261,0.1599])则tt1.size()为torch.Size([6]),是一个一行的tensor。现在通过view可以将其重构一下形状。result=tt1.view(3,2)resulttensor([[-0.3623.转载 2021-08-31 11:55:35 · 471 阅读 · 0 评论 -
torch.Tensor的4种乘法
torch.Tensor的4种乘法torch.Tensor有4种常见的乘法:*, torch.mul, torch.mm, torch.matmul. 本文抛砖引玉,简单叙述一下这4种乘法的区别,具体使用还是要参照官方文档。点乘a与b做*乘法,原则是如果a与b的size不同,则以某种方式将a或b进行复制,使得复制后的a和b的size相同,然后再将a和b做element-wise的乘法。下面以*标量和*一维向量为例展示上述过程。* 标量Tensor与标量k做*乘法的结果是Tensor的每个元转载 2021-08-10 11:51:40 · 1385 阅读 · 0 评论 -
ModuleList和Sequential
一、官方文档首先看官方文档的解释,仅列出了容器(Containers)中几个比较常用的CLASS。CLASS torch.nn.ModuleBase class for all neural network modules.Your models should also subclass this class.import torch.nn as nnimport torch.nn.functional as Fclass Model(nn.Module): def __转载 2021-03-22 18:52:35 · 477 阅读 · 0 评论 -
torch.cat() 和 torch.stack()
网上很多的示例,都在讨论二维数据(矩阵),单是对于做图像与深度学习的人来说均是三维起步,一般都是4维,下边以4维数据举例对于pytorch中的堆叠与拼接函数stack与cat,二者还是有一定的不同torch.cat这是一个拼接函数(姑且这么说)直接上例子a0=torch.Tensor([[[[1,1,1,1],[2,2,2,2]]]])a1=torch.Tensor([[[[3,3,3,3],[4,4,4,4]]]])torch.Size([1, 1, 2, 4])torch.ca转载 2021-03-03 23:27:07 · 789 阅读 · 1 评论 -
Pytorch中的model.train()和model.eval()
一、model.train()和model.eval()分别在训练和测试中都要写,它们的作用如下:(1)、model.train()启用BatchNormalization和Dropout,将BatchNormalization和Dropout置为True(2)、model.eval()不启用BatchNormalization和Dropout,将BatchNormalization和Dropout置为False二、区别如果模型中有BN层(Batch Normalizat...原创 2021-01-10 18:39:53 · 255 阅读 · 0 评论 -
torch.gather() 和torch.sactter_()的用法简析
https://blog.csdn.net/Teeyohuang/article/details/82186666?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-4.control&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-4.control转载 2021-01-03 19:44:50 · 140 阅读 · 0 评论 -
不同框架下的默认图片通道顺序
不同框架下的默认图片通道顺序caffe: NCHWpytorch: NCHWmxnet: NCHW海思bgr: NCHWNCNN: CHWtensorflow: NHWCopencv: NHWC瑞芯微rknn: NHWCscipy.misc: NHWC转载 2021-01-02 16:36:17 · 262 阅读 · 0 评论 -
[pytorch笔记] scatter_的使用
scatter_(input, dim, index, src)将src中数据根据index中的索引按照dim的方向填进input中.1 >>> x = torch.rand(2, 5)2 >>> x3 4 0.4319 0.6500 0.4080 0.8760 0.23555 0.2609 0.4711 0.8486 0.8573 0.10296 [torch.FloatTensor of size 2x5]1) dim =原创 2020-12-28 15:24:17 · 632 阅读 · 1 评论 -
tensor
原创 2020-12-17 11:54:51 · 80 阅读 · 0 评论