Python Nn.identity