class_docstr: EncoderLSTM Module wrapped in a lifted scan transform.
class_docstr_tok: ["EncoderLSTM","Module","wrapped","in","a","lifted","scan","transform","."]
code: def initialize_carry(batch_size: int, hidden_size: int):
      # Use a dummy key since the default state init fn is just zeros.
      return nn.LSTMCell(hidden_size, parent=None).initialize_carry(
          jax.random.PRNGKey(0), (batch_size, 1)
      )
  
imports: import functools
  from typing import Any
  from flax import linen
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 2 for calling the EncoderLSTM obj's underlying member method code with example usage: obj.initialize_carry(batch_size, hidden_size) and returns: nn