详解PyTorch中Tensor的高阶操作
yipeiwu_com6年前
条件选取：torch.where(condition, x, y) → Tensor 返回从 x 或 y 中选择元素的张量，取决于 condition 操作定义： 举个例子：...
PyTorch中Tensor的拼接与拆分的实现
yipeiwu_com6年前
拼接张量：torch.cat() 、torch.stack() torch.cat(inputs, dimension=0) → Tensor 在给定维度上对输入的张量序列 s...
PyTorch中Tensor的维度变换实现
yipeiwu_com6年前
对于 PyTorch 的基本数据对象 Tensor (张量)，在处理问题时，需要经常改变数据的维度，以便于后期的计算和进一步处理，本文旨在列举一些维度变换的方法并举例，方便大家查看。 维...
浅谈Pytorch中的torch.gather函数的含义
yipeiwu_com6年前
pytorch中的gather函数 pytorch比tensorflow更加编程友好，所以准备用pytorch试着做最近要做的一些实验。 立个flag开始学习pytorch，新开一个分类...
pytorch索引查找 index_select的例子
yipeiwu_com6年前
index_select anchor_w = self.FloatTensor(self.scaled_anchors).index_select(1, self.LongTensor...
pytorch打印网络结构的实例
yipeiwu_com6年前
最简单的方法当然可以直接print（net），但是这样网络比较复杂的时候效果不太好，看着比较乱；以前使用caffe的时候有一个网站可以在线生成网络框图，tensorflow可以用tens...
pytorch 更改预训练模型网络结构的方法
yipeiwu_com6年前
一个继承nn.module的model它包含一个叫做children()的函数，这个函数可以用来提取出model每一层的网络结构，在此基础上进行修改即可，修改方法如下(去除后两层)：...
关于pytorch多GPU训练实例与性能对比分析
yipeiwu_com6年前
以下实验是我在百度公司实习的时候做的，记录下来留个小经验。 多GPU训练 cifar10_97.23 使用 run.sh 文件开始训练 cifar10_97.50 使用 run.4GPU...
pytorch使用指定GPU训练的实例
yipeiwu_com6年前
本文适合多GPU的机器，并且每个用户需要单独使用GPU训练。 虽然pytorch提供了指定gpu的几种方式，但是使用不当的话会遇到out of memory的问题，主要是因为pytorc...
在pytorch中为Module和Tensor指定GPU的例子
yipeiwu_com6年前
pytorch指定GPU 在用pytorch写CNN的时候，发现一运行程序就卡住，然后cpu占用率100%，nvidia-smi 查看显卡发现并没有使用GPU。所以考虑将模型和输入数据及...
