Here is some sample code using PyTorch for generating a sample dataset using the MNIST dataset as an example: In this code, we first define the transformations to be applied…