Solved by ptrblck in post #2 
              
              
                Could you explain the use case a bit more, please? 
Would you like to randomly augment the sample/batch and concatenate these transformed versions to the current sample/batch? 
If so, I assume the batch size would be variable, since these transformations are applied randomly. 
In that case, I think …