Skip to content

Instantly share code, notes, and snippets.

@piEsposito

piEsposito/lstm-post-imports.py

Created May 24, 2020 22:02
Show Gist options
  • Save piEsposito/6727beb98e3101bab196b21da5ccf23d to your computer and use it in GitHub Desktop.
Save piEsposito/6727beb98e3101bab196b21da5ccf23d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
lstm-post-imports.py
import math
import torch
import torch.nn as nn
@fatalfeel
Copy link

ok i Got!~~~Done

i_t = torch.sigmoid(self.U_ix_t + self.ub_i + self.V_ih_t + self.vb_i)
f_t = torch.sigmoid(self.U_fx_t + self.ub_i + self.V_fh_t + self.vb_f)
g_t = torch.tanh( self.U_cx_t + self.ub_i + self.V_ch_t + self.vb_c)
o_t = torch.sigmoid(self.U_ox_t + self.ub_i + self.V_oh_t + self.vb_o)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment