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

Public Member Functions

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)
 

Public Attributes

 input_size
 
 hidden_size
 
 bias
 
 weight_ih
 
 weight_hh
 
 bias_ih
 
 bias_hh
 

Detailed Description

Definition at line 669 of file rnn.py.


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