You can use TensorDataset to create dataset from a tensor. You can split the 100x30x25x25 tensor into two and create two datasets for train and val.
Creating your dataset is also fairly transparent. Check out this tutorial: http://pytorch.org/tutorials/beginner/data_loading_tutorial.html 2.9k