for epoch_index in range(cfg.epoch):
for train_batch_index, (img_batch, label_batch) in enumerate(train_loader):
img_batch = variable(img_batch)
label_batch = variable(label_batch).unsqueeze(dim=1)
 img_batcg_s = torch.slice(img_batch)
 img_batcg_t = torch.slice(img_batch)

May I use this **torch.slice** to do this?