pub fn get_batch_for_gpts() -> Result<Tensor>
Helper function to a sample batch of tokens to feed into GPTs.