Hyeongu_Oh:

For example, I set the batch_size to 8, and I want to call each of the eight methods in one batch.


You could use a loop, if you want to call each value separately.
I don’t know, what set_model means, but if it cannot accept batched data, you would have to fallback to the loop (or use a batch size of 1).