tensorplay.nn.functional.lstm_cell

tensorplay.nn.functional.lstm_cell(input, hx, cx, w_ih, w_hh, b_ih=None, b_hh=None)[source]

One time step of a long short-term memory cell. Returns the next hidden state and next cell state.