Is there a way to classify a set of data as a whole via Pytorch?

Question

I\'m currently dealing with a classification task on a CT dataset. In CT datasets, multiple slices belong to one single patient, while setting up my dataset, I arrange my data a