tensorplay.nn.functional.gru_cell

tensorplay.nn.functional.gru_cell(input, hx, w_ih, w_hh, b_ih=None, b_hh=None) → Tensor[source]

One time step of a gated recurrent unit cell.