【pytorch】torch.nn.Identity()

identity模块不改变输入，直接return input

 

一种编码技巧吧，比如我们要加深网络，有些层是不改变输入数据的维度的，

在增减网络的过程中我们就可以用identity占个位置，这样网络整体层数永远不变，

应用：

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