pytorch学习之torch.normal()笔记

torch.noraml()

这里只讲一下常用的三个参数

torch.normal(mean, std, out=None) #得到的是一个张量(tensor)

介绍一下三个参数的意思:

mean:输出元素的正态分布均值(默认为float)

std:输出元素正态分布的标准差

out:输出张量的格式(默认1行)

注:meanstd的形状不需要匹配,但每个张量的元素总数需要相同。为什么呢??因为他这个是一一对应的。

# 输入(这是我刚开始学可以理解的)
import torch
torch.normal(2, 3, size=(2, 4))
# 输出如下
tensor([[ 9.6593,  5.6646, -5.6696,  3.8469],
        [ 2.4485,  2.6941,  4.3581,  0.4665]])

原本我以为我搜了资料后我觉得我理解了,结果出现了这个

# 输入(这个还是要把前面张量的学习好好加固一下)
torch.normal(mean=torch.arange(1., 11.), std=torch.arange(1, 0, -0.1))
# 输出(因为是随机数,每个人的结果可能不一样)
tensor([ 2.0341,  1.6641,  3.9183,  3.5596,  4.2670,  5.6417,  7.1381,  7.7588,
         9.2533, 10.0343])

好的,我搜了很久,单个搜normal我都没理解,后面我发现这个mean与std是一一对应的,我才发觉,这是一个一一对应的关系。

例如:

mean生成的是张量:tensor([ 1., 2., 3., 4., 5., 6., 7., 8., 9., 10.])

std生成的是张量:tensor([1.0000, 0.9000, 0.8000, 0.7000, 0.6000, 0.5000, 0.4000, 0.3000, 0.2000, 0.1000])

然后一一对应,默认out为一行(列数不定)。产生的第一个数字便是以1(mean产生张量的第一个元素)为正态分布的均值和std为1(std产生张量的第一个元素)的标准差产生的随机数。

  • 8
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
PyTorch是一个基于Python的科学计算库,主要针对深度学习任务。在PyTorch中,torch.nn是一个用于构建神经网络模型的模块。 torch.nn模块提供了一系列神经网络层和函数,方便用户构建自定义的神经网络。用户可以通过继承torch.nn.Module类来定义自己的神经网络模型。torch.nn模块中常用的类包括各种层(例如全连接层、卷积层、池化层和循环层等)、非线性激活函数和损失函数等。 在使用torch.nn模块构建神经网络时,用户需要实现模型的前向传播函数forward()。该函数定义了输入数据在神经网络中的流动方式,即通过层和函数的组合计算输出。在forward()函数中,用户可以使用已定义的层和函数进行计算,也可以实现自定义的操作。 torch.nn模块中的另一个重要概念是参数(parameter)。参数是模型中需要学习的变量,例如网络层的权重和偏置项。用户可以通过在模型中定义torch.nn.Parameter对象来创建参数,并在forward()函数中进行使用。 除了torch.nn模块外,PyTorch还提供了其他的工具和模块来辅助神经网络的训练和优化过程。例如torch.optim模块包含了各种优化算法,如随机梯度下降(SGD)、Adam等,用于更新模型中的参数。torch.utils.data模块提供了数据处理和加载的工具,方便用户使用自己的数据训练模型。 总之,torch.nn模块是PyTorch中用于构建神经网络模型的重要组成部分。通过使用torch.nn的各种类和函数,用户可以方便地创建自己想要的神经网络结构,并利用PyTorch强大的计算能力和优化算法来训练和优化模型。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值