Hello,
I would like to save some of my model outputs to file. This will be done only at the test step:
    def test_step(self, batch, batch_idx):
        x1, x2 = batch["x1"], batch["x2"]
        predicts = self(x1, x2) # which call model forward in eval mode
        save(predicts, predicts_path) # method to save predicts to file
      

What would be an efficient approach to concatenate in a file each batch predicts of the model (which has the shape [batch_size, hidden_size])?