Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | List of all members
torch.nn.modules.rnn.GRUCell Class Reference
Inheritance diagram for torch.nn.modules.rnn.GRUCell:
torch.nn.modules.rnn.RNNCellBase

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: