TH_NUM的博客

日常积累

PyTorch 常用方法总结4:张量维度操作(拼接、维度扩展、压缩、转置、重复……)

涉及的方法有下面几种 torch.cat() torch.Tensor.expand() torch.squeeze() torch.Tensor.repeat() torch.Tensor.narrow() torch.Tensor.view() torch.Tensor.r...

2018-10-16 15:09:56

阅读数:11

评论数:0

使用tuna等源安装pytorch和tensorflow

阿里云或者豆瓣的镜像。 阿里云 http://mirrors.aliyun.com/pypi/simple/ 中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/ 豆瓣 http://pypi.douban.com/simple/ 清华大学 https:/...

2018-09-21 17:07:52

阅读数:21

评论数:0

grad_fn 里面的next_functions 属性

next_functions 反向传播计算图的记录。 在反向图的计算中,计算图终止于叶子AccumulateGrad节点。有一个.variable属性指向叶子节点。 例子: a = torch.randn(1, requires_grad=True) b = a*(a+2) print ...

2018-09-05 10:31:29

阅读数:30

评论数:0

Pytorch常用函数解析(一) Tensor 拼接

torch模块下的数学操作符 1 . torch.numel() 返回一个tensor变量内所有元素个数,可以理解为矩阵内元素的个数 2 . torch.squeeze() 对于tensor变量进行维度压缩,去除维数为1的的维度。例如一矩阵维度为A*1*B*C*1*D,通过squeeze()返...

2018-08-23 13:30:21

阅读数:81

评论数:0

Pytorch变量类型转换

Pytorch的数据类型为各式各样的Tensor,Tensor可以理解为高维矩阵。与Numpy中的Array类似。Pytorch中的tensor又包括CPU上的数据类型和GPU上的数据类型,一般GPU上的Tensor是CPU上的Tensor加cuda()函数得到。通过使用Type函数可以查看变量类...

2018-08-20 14:10:55

阅读数:43

评论数:0

pytorch使用多GPU训练MNIST

下面的代码参数没有调试,可能准确率不高,仅仅供参考代码格式。 import argparse import torch import torch.nn as nn import torch.optim as optim import torch.nn.functional as F from...

2018-08-20 11:40:53

阅读数:128

评论数:0

tf.contrib.keras.preprocessing.sequence.pad_sequences 将标量数据 转换成numpy ndarray

keras.preprocessing.sequence.pad_sequences(sequences, maxlen=None, dtype=’int32’, padding=’pre’, truncating=’pre’, value=0.) 函数说明: 将长为nb_samples...

2018-07-03 21:50:53

阅读数:257

评论数:0

pytorch sampler对数据进行采样

PyTorch中还单独提供了一个sampler模块,用来对数据进行采样。常用的有随机采样器:RandomSampler,当dataloader的shuffle参数为True时,系统会自动调用这个采样器,实现打乱数据。默认的是采用SequentialSampler,它会按顺序一个一个进行采样。这里介...

2018-07-01 21:32:31

阅读数:686

评论数:0

pytorch 函数DataLoader

Dataset(https://blog.csdn.net/TH_NUM/article/details/80877196)只负责数据的抽象,一次调用getitem只返回一个样本。前面提到过,在训练神经网络时,最好是对一个batch的数据进行操作,同时还需要对数据进行shuffle和并行加速等。对...

2018-07-01 21:23:34

阅读数:710

评论数:0

pytorch Dataset 的ImageFolder

介绍常用的pytorch Dataset 里面的ImageFolder,实现和https://blog.csdn.net/TH_NUM/article/details/80877196很相似。 ImageFolder假设所有的文件按文件夹保存好,每个文件夹下面存贮同一类别的图片,文件夹的名字为...

2018-07-01 20:57:11

阅读数:1124

评论数:1

pytorch 数据处理:定义自己的数据集合

数据处理 版本1 #数据处理 import os import torch from torch.utils import data from PIL import Image import numpy as np #定义自己的数据集合 class DogCat(data.Dataset...

2018-07-01 20:33:02

阅读数:494

评论数:0

pytorch 函数clamp

clamp表示夹紧,夹住的意思,torch.clamp(input,min,max,out=None)-> Tensor 将input中的元素限制在[min,max]范围内并返回一个Tensor 用法:

2018-06-29 18:19:55

阅读数:1323

评论数:0

pytorch 常用函数 max ,eq

max找出tensor 的行或者列最大的值: 找出每行的最大值: import torch outputs=torch.FloatTensor([[1],[2],[3]]) print(torch.max(outputs.data,1)) 输出: (tensor([ 1., 2.,...

2018-06-23 13:56:53

阅读数:715

评论数:0

0.4.0pytorch运行过程中对0-dim和volatile提示UserWarning的解决方法

将pytorch更新到0.4.0最新版后对0.3.1版本代码会有如下警告,它在提醒用户下个版本这将成为一个错误 UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use t...

2018-06-23 13:25:25

阅读数:633

评论数:0

提示
确定要删除当前文章?
取消 删除
关闭
关闭