Skip to content

Instantly share code, notes, and snippets.

@lazuxd

lazuxd/deep_gru_lstm_call_level.py

Created July 20, 2021 12:54
Show Gist options
  • Save lazuxd/9e4355f027f7af360b262c3b59edc0d7 to your computer and use it in GitHub Desktop.
Save lazuxd/9e4355f027f7af360b262c3b59edc0d7 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
deep_gru_lstm_call_level.py
def _call_level(self,
level: int,
x: Union[np.ndarray, tf.Tensor]) -> tf.Tensor:
return (self._call_gru(level, x) if self.unit_type == 'gru'
else self._call_lstm(level, x))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment