MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / _stack_h_n

Method _stack_h_n

imperative/python/megengine/module/rnn.py:290–291  ·  view source on GitHub ↗
(self, h_n)

Source from the content-addressed store, hash-verified

288 return fn(hx)
289
290 def _stack_h_n(self, h_n):
291 return stack(h_n, axis=0)
292
293 def forward(self, input: Tensor, hx=None):
294 if self.batch_first:

Callers

nothing calls this directly

Calls 1

stackFunction · 0.50

Tested by

no test coverage detected