科学网

 找回密码
  注册
搜索
热搜: 科学 论文
搜索
[Pytorch函数] .masked_fill_()与Numpy数组中None的作用
张伟 2020-11-19 16:42
(一).masked_fill_() masked_fill_(mask, value) 掩码操作 用value填充tensor中与mask中值为1位置相对应的元素。mask的形状必须与要填充的tensor形状一致。 a = torch.randn(5,6) x = mask = torch.zeros(5,6,dtype=torch.float) for e_id, src_len in enumerate(x): mask = 1 ...
个人分类: Pytorch|372 次阅读|没有评论
import torchnet as tnt
张伟 2020-11-13 16:49
主要介绍torchnet中的常用模块方法 github网址: https://github.com/facebookarchive/torchnet import torchnet as tnt acc_meter = tnt.meter.ClassErrorMeter(accuracy=True) #计算精度 loss_meter = tnt.meter.AverageValueMeter() #计算存储数据的均值和标准差 (1)acc_meter使用示例 pred=torch.Tensor( ...
个人分类: Pytorch|494 次阅读|没有评论
Pytorch采样器详解--结合DataLoader使用
张伟 2020-11-9 16:21
Pytorch采样器详解--结合DataLoader使用,产生选取样本的随机编号,用于抽样!! traindataloader = torch.utils.data.DataLoader(dataset=traindataset, sampler= RandomSampler(traindataset) ,   ...
个人分类: Pytorch|319 次阅读|没有评论
Pytorch中torch.autograd.grad()函数用法示例
张伟 2020-11-7 10:43
一、函数解释 如果输入x,输出是y,则求y关于x的导数(梯度): def grad(outputs, inputs, grad_outputs=None, retain_graph=None, create_graph=False, only_inputs=True, allow_unused=False): rComputes and returns the sum of gradients of outputs ...
个人分类: Pytorch|417 次阅读|没有评论
PyTorch的torch.cat
张伟 2020-10-29 19:56
1. 字面理解:torch.cat是将两个张量(tensor)拼接在一起, cat是concatnate的意思,即拼接 ,联系在一起。 2. 例子理解 import torch A=torch.ones(2,3) #2x3的张量(矩阵) & ...
个人分类: Pytorch|457 次阅读|没有评论
pytorch中contiguous()
张伟 2020-10-29 09:04
contiguous:view只能用在contiguous的variable上。如果在view之前用了transpose, permute等,需要用contiguous()来返回一个contiguous copy。 一种可能的解释是: 有些tensor并不是占用一整块内存,而是由不同的数据块组成,而 tensor的view()操作依赖于内存是整 ...
个人分类: Pytorch|354 次阅读|没有评论
Pytorch中LayerNorm计算过程详解
张伟 2020-10-28 22:52
【参考】 https://blog.csdn.net/weixin_39228381/article/details/107939602 点滴分享,福泽你我!Addoil!
个人分类: Pytorch|649 次阅读|没有评论
Pytorch中DataLoader与DataSet联合使用
张伟 2020-10-28 16:24
DataLoader与DataSet关系 DataLoader是Pytorch用来 加载数据 的一个类,其实就是一个迭代器,而迭代的数据从哪来?就需要用到DataSet了。 DataSet就是用来 封装数据 的类,主要用来对数据进行相关的自定义操作(比如图片的裁剪、标签的定义等),通过 __getitem__ ...
个人分类: Pytorch|387 次阅读|没有评论
DataLoader使用完整流程及数据格式
张伟 2020-10-27 19:55
通过一个小案例,反映出DataLoader使用完整流程!! import torch import torch.utils.data as Data BATCH_SIZE = 3 #批训练数据个数 x = torch.linspace(1,10,10) #x data (torch tensor) y = torch.linspace(10,1,10) #y data (torch tensor) #随后我们需要把X和Y组成一个完整的数据集,并转 ...
个人分类: Pytorch|586 次阅读|没有评论

本页有 1 篇博文因作者的隐私设置或未通过审核而隐藏

Archiver|手机版|科学网 ( 京ICP备07017567号-12 )

GMT+8, 2020-11-24 12:27

Powered by ScienceNet.cn

Copyright © 2007- 中国科学报社

返回顶部