Identity関数をちゃんと用意してくれてるフレームワークは好きだよ。
https://pytorch.org/docs/stable/generated/torch.nn.Identity.html
Experimental private instance. Running on FreeBSD!