Pytorch加载部分预训练模型的参数实例
yipeiwu_com6年前
前言 自从从深度学习框架caffe转到Pytorch之后，感觉Pytorch的优点妙不可言，各种设计简洁，方便研究网络结构修改，容易上手，比TensorFlow的臃肿好多了。对于深度学习...
pytorch中的embedding词向量的使用方法
yipeiwu_com6年前
Embedding 词嵌入在 pytorch 中非常简单，只需要调用 torch.nn.Embedding(m, n) 就可以了，m 表示单词的总数目，n 表示词嵌入的维度，其实词嵌入就...
对Pytorch神经网络初始化kaiming分布详解
yipeiwu_com6年前
函数的增益值 torch.nn.init.calculate_gain(nonlinearity, param=None) 提供了对非线性函数增益值的计算。 增益值gain是一个...
在PyTorch中Tensor的查找和筛选例子
yipeiwu_com6年前
本文源码基于版本1.0，交互界面基于0.4.1 import torch 按照指定轴上的坐标进行过滤 index_select() 沿着某tensor的一个轴dim筛选若干个坐标 &...
Pytorch Tensor的索引与切片例子
yipeiwu_com6年前
1. Pytorch风格的索引 根据Tensor的shape，从前往后索引，依次在每个维度上做索引。 示例代码： import torch a = torch.rand(4, 3...
PyTorch之图像和Tensor填充的实例
yipeiwu_com6年前
在PyTorch中可以对图像和Tensor进行填充，如常量值填充，镜像填充和复制填充等。在图像预处理阶段设置图像边界填充的方式如下： import vision.torchvisio...
Pytorch实现GoogLeNet的方法
yipeiwu_com6年前
GoogLeNet也叫InceptionNet，在2014年被提出，如今已到V4版本。GoogleNet比VGGNet具有更深的网络结构，一共有22层，但是参数比AlexNet要少12倍...
浅析PyTorch中nn.Linear的使用
yipeiwu_com6年前
查看源码 Linear 的初始化部分： class Linear(Module): ... __constants__ = ['bias'] def __init__(s...
详解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...
