PyTorch——add_module

最新推荐文章于 2024-05-12 09:01:33 发布

倔强一撮毛

最新推荐文章于 2024-05-12 09:01:33 发布

阅读量465

点赞数

分类专栏： PyTorch学习文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/weixin_41658139/article/details/128412253

版权

PyTorch学习专栏收录该内容

2 篇文章 1 订阅

订阅专栏

torch.nn.Module.add_module

添加一个子模块到当前的模块。

模块可以用给定的名称以属性获取。

输入参数有：

name：str类型，子模块名称；
module：nn.Module类型，添加到模块的子模块；

import torch
from torch import nn

o_module = nn.Module()
print(o_module)  # Module()

o_module.add_module('my_conv2d', nn.Conv2d(3, 16, 3))
print(o_module)  # Module(((mv_conv2d): Conv2d(3, 16, kernel_size=(3, 3), stride=(1, 1))

img = torch.randn(1, 3, 256, 256)
print(o_module.my_conv2d(img.shape))  # torch.Size([1, 16, 254, 254])