Pytorch Sequential Add Layer