 |
Caffe2 - Python API
A deep learning, cross platform ML framework
|
|
Caffe2 - Python API
A deep learning, cross platform ML framework
|
Public Member Functions | |
| def | __init__ (self, input_size, hidden_size, forget_bias, memory_optimization, drop_states=False, linear_before_reset=False, kwargs) |
| def | prepare_input (self, model, input_blob) |
| def | get_state_names (self) |
| def | get_output_dim (self) |
 Public Member Functions inherited from caffe2.python.rnn_cell.RNNCell | |
| def | __init__ (self, name=None, forward_only=False, initializer=None) |
| def | initializer (self) |
| def | initializer (self, value) |
| def | scope (self, name) |
| def | apply_over_sequence (self, model, inputs, seq_lengths=None, initial_states=None, outputs_with_grads=None) |
| def | apply (self, model, input_t, seq_lengths, states, timestep) |
| def | apply_override (self, model, input_t, seq_lengths, timestep, extra_inputs=None) |
| def | prepare_input (self, model, input_blob) |
| def | get_output_state_index (self) |
| def | get_state_names (self) |
| def | get_state_names_override (self) |
| def | get_output_dim (self) |
Public Attributes | |
| input_size | |
| hidden_size | |
| forget_bias | |
| memory_optimization | |
| drop_states | |
| linear_before_reset | |
| Public Attributes inherited from caffe2.python.rnn_cell.RNNCell | |
| name | |
| recompute_blobs | |
| forward_only | |
Definition at line 10 of file gru_cell.py.
1.8.11
