I\'m training a data set with 17 features and 5 output using pytorch. But I\'m most interested in two of them, let say output 2 and 3 out of 0-4. What\'s a good strategy to