网络编程
位置:首页>> 网络编程>> Python编程>> python中torch.nn.identity()方法详解

python中torch.nn.identity()方法详解

作者:sigmoidAndRELU  发布时间:2021-05-21 16:51:01 

标签:python,torch.nn.identity()

先看代码

m = nn.Identity(
54,
unused_argument1=0.1,
unused_argument2=False
)

input = torch.randn(128, 20)
output = m(input)
>>> print(output.size())
torch.Size([128, 20])

这是官方文档中给出的代码,很明显,没有什么变化,输入的是torch,输出也是,并且给定的参数似乎并没有起到变化的效果。

看源码

class Identity(Module):
   r"""A placeholder identity operator that is argument-insensitive.

Args:
       args: any argument (unused)
       kwargs: any keyword argument (unused)

Examples::

>>> m = nn.Identity(54, unused_argument1=0.1, unused_argument2=False)
       >>> input = torch.randn(128, 20)
       >>> output = m(input)
       >>> print(output.size())
       torch.Size([128, 20])

"""
   def __init__(self, *args, **kwargs):
       super(Identity, self).__init__()

def forward(self, input: Tensor) -> Tensor:
       return input

这相当的简洁明了啊,输入是啥,直接给输出,不做任何的改变。再看文档中的一句话:A placeholder identity operator that is argument-insensitive.

翻译一下就是:不区分参数的占位符标识运算符。百度翻译,其实意思就是这个网络层的设计是用于占位的,即不干活,只是有这么一个层,放到残差网络里就是在跳过连接的地方用这个层,显得没有那么空虚!

应用

例如此时:如果此时我们使用了se_layer,那么就SELayer(dim),否则就输入什么就输出什么(什么都不做)

python中torch.nn.identity()方法详解

python中torch.nn.identity()方法详解

来源:https://blog.csdn.net/TTLoveYuYu/article/details/118224298

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com