gated_rnn.py 文件源码

def __init__(self, cell, *args, **kwargs):
      """Construct the CellWithExtraInput.

      Args:
        cell: The RNNCell to wrap (typically a nn.RNNCore).
        *args: Extra arguments to pass to __call__.
        **kwargs: Extra keyword arguments to pass to __call__.
      """
      self._cell = cell
      self._args = args
      self._kwargs = kwargs