PyTorch学习问题记录

Q1：def train() 中的model.train()的作用是什么？为什么要写？

A1：class torch.nn.Module中 train(mode=True)

　　Sets the module in training mode. This has any effect only on modules such as Dropout or BatchNorm.

　　参看 http://pytorch.org/docs/master/nn.html

Q2：torch.gather()函数的功能是什么？

 1 t = torch.Tensor([[1, 2], [3, 4]])
 2 print(t)
 3 a = torch.gather(t, 1, torch.LongTensor([[0,0], [1,0]]))
 4 print(a)
 5 '''
 6  1  2
 7  3  4
 8 [torch.FloatTensor of size 2x2]
 9
10  1  1
11  4  3
12 [torch.FloatTensor of size 2x2]
13 '''

A2：

out[i][j][k] = input[index[i][j][k]][j][k] 　　# if dim == 0
out[i][j][k] = input[i][index[i][j][k]][k] 　　# if dim == 1
out[i][j][k] = input[i][j][index[i][j][k]] 　　# if dim == 2

out[i][j] = input[index[i][j]][j]
out[i][j] = input[i][index[i][j]]

out[0][0] = input[0][index[0][0]] = input[0][0] = 1
out[0][1] = input[0][index[0][1]] = input[0][0] = 1
out[1][0] = input[1][index[1][0]] = input[1][1] = 4

out[1][1] = input[1][index[1][1]] = input[1][0] = 3

Q3：torch.norm() 函数的功能是什么？

 1 a = torch.FloatTensor([[1, 2], [3, 4]])
 2 b = torch.norm(a)
 3 print(a)
 4 print(b)
 5 '''
 6  1  2
 7  3  4
 8 [torch.FloatTensor of size 2x2]
 9
10 5.477225575051661
11 '''

A3：

norm() 函数是求范数，一般默认是2范数。平方和开根号。

参考博文：几种范数的简单介绍

normal() 函数是求正太分布。

Q4: topk()函数

torch.Tensor.topk (Python method, in torch.Tensor) ||topk(k, dim=None, largest=True, sorted=True) -> (Tensor, LongTensor)
torch.topk (Python function, in torch) ||torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor)

1 topi = torch.LongTensor([5])        # [torch.LongTensor of size 1]
2 topii = torch.LongTensor([[5]])     # [torch.LongTensor of size 1x1] 3 ni = topi[0] 4 nii = topii[0][0] 5 print(ni, nii) # 5 5

Q5:

1 loss = Variable(torch.FloatTensor([1]))
2 print(loss.data)                      # 1 [torch.FloatTensor of size 1]
3 print(loss.data[0])                   # 1.0

转载于:https://www.cnblogs.com/Joyce-song94/p/7252206.html

PyTorch学习问题记录相关推荐

Pytorch学习记录-torchtext和Pytorch的实例（使用神经网络训练Seq2Seq代码）
Pytorch学习记录-torchtext和Pytorch的实例1 0. PyTorch Seq2Seq项目介绍 1. 使用神经网络训练Seq2Seq 1.1 简介,对论文中公式的解读 1.2 数据预 ...
神经网络学习小记录52——Pytorch搭建孪生神经网络（Siamese network）比较图片相似性
神经网络学习小记录52--Pytorch搭建孪生神经网络(Siamese network)比较图片相似性学习前言什么是孪生神经网络代码下载孪生神经网络的实现思路一.预测部分 1.主干网络介绍 ...
神经网络学习小记录50——Pytorch 利用efficientnet系列模型搭建yolov3目标检测平台
神经网络学习小记录50--Pytorch 利用efficientnet系列模型搭建yolov3目标检测平台学习前言什么是EfficientNet模型源码下载 EfficientNet模型的实现思 ...
PyTorch学习记录——PyTorch进阶训练技巧
PyTorch学习记录--PyTorch进阶训练技巧 1.自定义损失函数 1.1 以函数的方式定义损失函数 1.2 以类的方式定义损失函数 1.3 比较与思考 2.动态调整学习率 2.1 官方提供的s ...
神经网络学习小记录69——Pytorch 使用Google Colab进行深度学习
神经网络学习小记录69--Pytorch 使用Google Colab进行深度学习注意事项学习前言什么是Google Colab 相关链接利用Colab进行训练一.数据集与预训练权重的上传 ...
pytorch 学习中安装的包
记录pytorch学习遇到的包 1.ImportError: cannot import name 'PILLOW_VERSION' torchvision 模块内import pillow的时候发现 ...
2_初学者快速掌握主流深度学习框架Tensorflow、Keras、Pytorch学习代码（20181211）
初学者快速掌握主流深度学习框架Tensorflow.Keras.Pytorch学习代码一.TensorFlow 1.资源地址: 2.资源介绍: 3.配置环境: 4.资源目录: 二.Keras 1.资 ...
Pytorch学习-tensorboard的使用
Pytorch学习-tensorboard的使用 1 Tensorboard简介运行机制安装及测试 2 SummaryWriter实例的使用教程 (1)初始化summaryWriter的方法 (2 ...
Pytorch学习-Task1
PyTorch学习-Task1:PyTorch张量计算与Numpy的转换张量 Tensor 1.张量的定义 2.张量的运算 3.Tensor与Numpy类型的转换 4.自动求导 pytorch 与神 ...

PyTorch学习问题记录

PyTorch学习问题记录相关推荐

最新文章

热门文章