title: How to fit custom data into Pytorch DataLoader?
body: <p>I have pre-processed and normalized my data, and split into training set and testing set. I have the following dimensions for my x_train and y_train:
  Shape of X_Train:  (708, 256, 3)
  Shape of Y_Train:  (708, 4)</p>
  
  <p>As you can see, x_train is 3-D. How can I go about inputting it into the pytorch dataloader? What do I put for the class block?</p>
  
  <pre><code>class training_set(Dataset):
      def __init__(self,X,Y):
  
      def __len__(self):
          return 
  
      def __getitem__(self, idx):
          return 
  
  training_set = torch.utils.data.TensorDataset(x_train, y_train)
  train_loader = torch.utils.data.DataLoader(training_set, batch_size=50, shuffle=True)
  </code></pre>