torch.nn.modules.rnn.GRUCell Class Reference

Inheritance diagram for torch.nn.modules.rnn.GRUCell:

Public Member Functions
def	__init__ (self, input_size, hidden_size, bias=True)
def	forward (self, input, hx=None)
Public Member Functions inherited from torch.nn.modules.rnn.RNNCellBase
def	__init__ (self, input_size, hidden_size, bias, num_chunks)
def	extra_repr (self)
def	check_forward_input (self, input)
def	check_forward_hidden (self, input, hx, hidden_label='')
def	reset_parameters (self)

Additional Inherited Members
Public Attributes inherited from torch.nn.modules.rnn.RNNCellBase
	input_size
	hidden_size
	bias
	weight_ih
	weight_hh
	bias_ih
	bias_hh

Detailed Description

Definition at line 893 of file rnn.py.

---

The documentation for this class was generated from the following file:

• torch/nn/modules/rnn.py