I wanted to add lables and channels of an image seprately by myself?

Question

for performing linear regression i did torch.stack and torch.cat to every other the torch.cat operation it took me about 10 min so i wanted to konw an optimal approa