Tensorflow 在pytorch模块中使用ImagedataGenerator

Tensorflow 在pytorch模块中使用ImagedataGenerator,tensorflow,keras,pytorch,cnn,Tensorflow,Keras,Pytorch,Cnn,我可以在keras中使用imagedatagenerator来训练pytorch cnn模块吗??如果没有,pytorch中是否有其他关键字具有相同的功能?torchvision.transforms?

我可以在keras中使用
imagedatagenerator
来训练pytorch cnn模块吗??如果没有,pytorch中是否有其他关键字具有相同的功能?

torchvision.transforms