hub / github.com/dailenson/SDT / forward_pre

Method forward_pre

models/transformer.py:232–253 · view source on GitHub ↗

(self, tgt, memory,
                    tgt_mask: Optional[Tensor] = None,
                    memory_mask: Optional[Tensor] = None,
                    tgt_key_padding_mask: Optional[Tensor] = None,
                    memory_key_padding_mask: Optional[Tensor] = None,
                    pos: Optional[Tensor] = None,
                    query_pos: Optional[Tensor] = None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

forwardMethod · 0.95

Calls 1

with_pos_embedMethod · 0.95

Tested by

no test coverage detected