Solved by mhubii in post #2 
              
              
                Hi @James_Trueb 
these transforms are ment to be applied to datasets. For example have a look at the MNIST example. You will find this line, 
auto test_dataset = torch::data::datasets::MNIST(
                        kDataRoot, torch::data::datasets::MNIST::Mode::kTest)
                        .map(t…