How can I extend into other datasets such as MINIST?

[HelenMao]

Hi, thanks for sharing your codes! I notice your codes you embed the gaussian gird data in Bourgansampling directly when generating the gaussian grid. If I want to extend into other image datasets such as MINIST, how can I embed the data in Bourgansampling? Looking forward to your answer, thanks!