pytorch python中torch.nn.identity()方法

【省流】nn.identity就是一个占位层,不知道占位层概念的小伙伴可以联想到“占位符”,没错,nn.identity的作用和占位符一样,只空占一个位置,啥活也不干。
【官方说明】
pytorch python中torch.nn.identity()方法_第1张图片
【官方例子】

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])

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

翻译一下就是:不区分参数的占位符标识运算符。其实意思就是这个网络层的设计是用于占位的,即不干活,只是有这么一个层,放到残差网络里就是在跳过连接的地方用这个层,显得没有那么空虚!
【实例】
例如:如果我们使用了se_layer,那么就SELayer(dim),否则输入什么就输出什么(什么都不做)
pytorch python中torch.nn.identity()方法_第2张图片
pytorch python中torch.nn.identity()方法_第3张图片

你可能感兴趣的:(python,pytorch,深度学习)