nn_module.py

最新推荐文章于 2026-01-24 01:11:09 发布

原创最新推荐文章于 2026-01-24 01:11:09 发布 · 340 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#pytorch

收录于

深度学习专栏收录该内容

51 篇文章

订阅专栏

本文介绍了如何使用PyTorch创建一个基础的模块，并通过实例演示了如何向前传播输入并实现加一操作。读者将学习到nn.Module的基本用法和模型的前向传播过程。

import torch
from torch import nn


class Tudui(nn.Module):
    def __init__(self):
        super().__init__()

    def forward(self, input):
        output = input + 1
        return output

tudui = Tudui()
x = torch.tensor(1.0)
output = tudui(x)
print(output)