this wraps an RNN cell to make it output its concatenated state instead of the output More...
Inheritance diagram for nabu.neuralnetworks.components.rnn_cell.StateOutputWrapper:
Public Member Functions | |
| def | __init__ (self, cell) |
| StateOutputWrapper constructo. | |
| def | output_size (self) |
| return cell output size | |
| def | state_size (self) |
| return cell state size | |
| def | zero_state (self, batch_size, dtype) |
| the cell zero state | |
| def | __call__ (self, inputs, state, scope=None) |
| call wrapped cell with constant scope | |
Detailed Description
this wraps an RNN cell to make it output its concatenated state instead of the output
The documentation for this class was generated from the following file:
