Torch nn flatten. .
Torch nn flatten .