Need help to visualize some tensors from one pytorch project

i was following one pytorch project, some part of that project is difficult for me to understand. so i am looking for expert who help me to understand and visualize. i was following [login to view URL] work. This is the pytorch implementation of "[login to view URL]" work. long story short, it took 800 normal and 810 abnormal video for training. First I want to visualize it's normal and abnormal video after passed through C3D, to do that i tried to plot normal_segments_scores from line 43 "[login to view URL] " in there i can get 30 video information (for training they took one batch 30 video), but problem is i can't get other batch information, means i can't reach other 770 videos informations.