MCPcopy Create free account
hub / github.com/pytorch/pytorch / _prepare_output

Method _prepare_output

caffe2/python/rnn_cell.py:856–863  ·  view source on GitHub ↗
(self, model, states)

Source from the content-addressed store, hash-verified

854 )
855
856 def _prepare_output(self, model, states):
857 output = self.internal_cell._prepare_output(
858 model,
859 states,
860 )
861 if self.dropout_ratio is not None:
862 output = self._apply_dropout(model, output)
863 return output
864
865 def _prepare_output_sequence(self, model, state_outputs):
866 output = self.internal_cell._prepare_output_sequence(

Callers

nothing calls this directly

Calls 2

_apply_dropoutMethod · 0.95
_prepare_outputMethod · 0.45

Tested by

no test coverage detected