Nn linear in pytorch.