 |
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, args, kwargs) |
| def | check_forward_args (self, input, hidden, batch_sizes) |
| def | permute_hidden (self, hx, permutation) |
| def | forward_impl (self, input, hx, batch_sizes, max_batch_size, sorted_indices) |
| def | forward_tensor (self, input, hx=None) |
| def | forward_packed (self, input, hx=None) |
| def | forward (self, input, hx=None) |
 Public Member Functions inherited from torch.nn.modules.rnn.RNNBase | |
| def | __init__ (self, mode, input_size, hidden_size, num_layers=1, bias=True, batch_first=False, dropout=0., bidirectional=False) |
| def | flatten_parameters (self) |
| def | reset_parameters (self) |
| def | get_flat_weights (self) |
| def | check_input (self, input, batch_sizes) |
| def | get_expected_hidden_size (self, input, batch_sizes) |
| def | check_hidden_size (self, hx, expected_hidden_size, msg='Expected hidden size{}, got) |
| def | check_forward_args (self, input, hidden, batch_sizes) |
| def | permute_hidden (self, hx, permutation) |
| def | forward (self, input, hx=None) |
| def | extra_repr (self) |
| def | __setstate__ (self, d) |
| def | all_weights (self) |
Additional Inherited Members | |
| Public Attributes inherited from torch.nn.modules.rnn.RNNBase | |
| mode | |
| input_size | |
| hidden_size | |
| num_layers | |
| bias | |
| batch_first | |
| dropout | |
| bidirectional | |
1.8.11
