python pytorch自定义_PyTorch使用自定义模块创建数据模型

2024-04-12 17:46:07

还有另一种查找预测的方法。在上一节中, 我们使用forward()和实现线性模型来找到预测。此方法非常有效且可靠。很容易理解和实施。

在”定制模块”中, 我们使用类创建一个定制模块, 它是init()和forward()方法和模型。 init()方法用于初始化类的新实例。在此init()方法中, 第一个参数是self, 它指示该类的实例(该对象尚未初始化), 而在其自身之后, 我们可以添加其他参数。

下一个参数是初始化线性模型的实例。在上一节中, 初始化线性模型需要输入大小以及输出大小等于1, 但是在自定义模块中, 我们传递输入大小和输出大小变量而不传递其默认值。

在这种情况下, 需要导入割炬的nn包。在此, 我们使用继承, 以便此子类将利用我们的基类和模块中的代码。

模块本身通常将充当所有神经网络模块的基类。之后, 我们创建一个模型以进行预测。

让我们看一个示例, 该示例如何通过创建自定义模块来完成预测：

对于单个数据

import torch

import torch.nn as nn

class LinearRegression(nn.Module):

def __init__(self, input_size, output_size):

super().__init__()

self.linear=nn.Linear(input_size, output_size)

def forward(self, x):

pred=self.linear(x)

return pred

torch.manual_seed(1)

model=LinearRegression(1, 1)

x=torch.tensor([1.0])

print(model.forward(x))

输出

tensor([0.0739], grad_fn=)

对于多个数据

import torch

import torch.nn as nn

class LinearRegression(nn.Module):

def __init__(self, input_size, output_size):

super().__init__()

self.linear=nn.Linear(input_size, output_size)

def forward(self, x):

pred=self.linear(x)

return pred

torch.manual_seed(1)

model=LinearRegression(1, 1)

x=torch.tensor([[1.0], [2.0], [3.0]])

print(model.forward(x))

输出

tensor([[0.0739], [0.5891], [1.1044]], grad_fn=)

python pytorch自定义_PyTorch使用自定义模块创建数据模型相关推荐

python pytorch自定义_Pytorch 实现自定义参数层的例子
注意,一般官方接口都带有可导功能,如果你实现的层不具有可导功能,就需要自己实现梯度的反向传递. 官方Linear层: class Linear(Module): def __init__(self, ...
python pytorch语音识别_PyTorch通过ASR实现语音到文本端的模型以及pytorch语音识别(speech) - pytorch中文网...
ASR,英文的全称是Automated Speech Recognition,即自动语音识别技术,它是一种将人的语音转换为文本的技术.今天我们主要了解pytorch实现语音到文本的端到端模型. spe ...
【pytorch速成】Pytorch图像分类从模型自定义到测试
文章首发于微信公众号<与有三学AI> [pytorch速成]Pytorch图像分类从模型自定义到测试前面已跟大家介绍了Caffe和TensorFlow,链接如下. [caffe速成]ca ...
python调用mysql中的自定义function并且返回结果
首先在Mysql中创立一个叫ForeignKeyTest的function,链接如下: https://blog.csdn.net/appleyuchi/article/details/7945291 ...
Python with上下文管理及自定义上下文管理
Python with上下文管理及自定义上下文管理一.上下文管理器上下文管理器是对Context Manager的翻译 ,上下文是 context 直译的叫法,在程序中用来表示代码执行过程中所处的 ...
禅道设置bug模板_禅道的自定义功能:表格切换和创建页面的自定义
原标题:禅道的自定义功能:表格切换和创建页面的自定义禅道8.2beta版本开始,增强了自定义功能. 用户可以根据自己的使用需要个性化设置禅道页面所显示的功能模块和字段. 可自定义的项如下: 导航菜单 ...
Pytorch学习笔记7——自定义数据集
Pytorch学习笔记7--自定义数据集 1.读取数据首先继承自torch.utils.data.Dataset 重写len与getitem train就用train数据集,test就用test数据 ...
Android流媒体开发之自定义一个完备的log模块
Android音视频开发之自定义一个完备的log模块前言基础知识的掌握 Log系统为什么需要自定义一个log模块呢? 做什么? 怎么做? 确定成员变量: 初始化LogUtil 输出功能的实现 ...
筛选服务器连接暂时不可用,某个自定义筛选器或模块khy-016(如 URLScan)限制了对该文件的访问...
摘要:服务器技术网nifengsoft 某个自定义筛选器或模块khy-016(如 URLScan)限制了对该文件的访问HTTP 错误 404.0 - Not Found 您要找的资源已被删除.已更名或 ...

最新文章

热门文章