Hi. As far as I understand your figure, the red box below is the Cell
.
That is the reason Cell return
h_1 (batch, hidden_size): tensor containing the next hidden state for each element in the batch
c_1 (batch, hidden_size): tensor containing the next cell state for each element in the batch