Write a pytorch linear gapfilling module