Exactly, 64 is the batch size, and 76,800 is the temporal size. But could not get what you mean by using the last time step only? Actually, I’ve added the whole source code in the OP. Could you please guide me on it?