API symbolsnnlatest (dev)latest (dev)2026-09-08v1.0.02026-09-06CopyCopy as MarkdownCopy page linkView MarkdownOpenAIOpen in ChatGPTClaudeOpen in ClaudePerplexityOpen in PerplexityGrokOpen in Grok 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. Help improve this pageFound an error, an unclear step, or a missing example?Was this page helpful?YesNoSuggest an editReport an issuetensorplay.nn.functional.grouped_mmPrevious Pagetensorplay.nn.functional.gumbel_softmaxSample from the Gumbel-Softmax distribution and optionally discretize.
