MCPcopy Create free account

hub / github.com/Lancelot39/KGSF / functions

Functions130 in github.com/Lancelot39/KGSF

Methodforward
(self, h, mask)
models/graph.py:156
Methodforward
(self, h)
models/graph.py:181
Methodforward
(self, input, memory, mask=None)
models/graph.py:201
Methodforward
(self, x, adj)
models/graph.py:232
Methodforward
(ctx, indices, values, shape, b)
models/graph.py:242
Methodforward
(self, indices, values, shape, b)
models/graph.py:262
Methodforward
(self, input, adj)
models/graph.py:288
Methodforward
(self, x, adj)
models/graph.py:355
Methodforward
(self, xs, ys)
models/transformer.py:134
Methodforward
(self, query, key=None, value=None, mask=None)
models/transformer.py:168
Methodforward
(self, x)
models/transformer.py:242
Methodforward
(self, *args)
models/transformer.py:260
Methodforward
input data is a FloatTensor of shape [batch, seq_len, dim] mask is a ByteTensor of shape [batch, seq_len], filled with 1 wh
models/transformer.py:340
Methodforward
(self, tensor, mask)
models/transformer.py:388
Methodforward
input data is a FloatTensor of shape [batch, seq_len, dim] mask is a ByteTensor of shape [batch, seq_len], filled with 1 wh
models/transformer.py:491
Methodforward
input data is a FloatTensor of shape [batch, seq_len, dim] mask is a ByteTensor of shape [batch, seq_len], filled with 1 wh
models/transformer.py:613
Methodforward
(self, x, encoder_output, encoder_mask)
models/transformer.py:671
Methodforward
(self, input, encoder_state, incr_state=None)
models/transformer.py:786
Methodforward
(self, x, encoder_output, encoder_mask, kg_encoder_output, kg_encoder_mask, db_encoder_output, db_encoder_mask
models/transformer.py:841
Methodforward
(self, input, encoder_state, encoder_kg_state, encoder_db_state, incr_state=None)
models/transformer.py:978
Methodforward
(self, xs, mems, cands)
models/transformer.py:1094
Methodforward
Get output predictions from the model. :param xs: input to the encoder :type xs: LongTensor[bs
models/transformer.py:1293
Functionis_distributed
Returns True if we are in distributed mode.
run.py:48
Methodprepare_head
(tensor)
models/transformer.py:179
Methodprepare_word2vec
(self)
dataset.py:52
Methodreorder_decoder_incremental_state
Reorder incremental state for the decoder. Used to expand selected beams in beam_search. Unlike reorder_encoder_states,
model.py:500
Methodreorder_decoder_incremental_state
Reorder incremental state for the decoder. Used to expand selected beams in beam_search. Unlike reorder_encoder_states,
models/transformer.py:1259
Methodreorder_encoder_states
Reorder encoder states according to a new set of indices. This is an abstract method, and *must* be implemented by the user.
model.py:447
Methodreorder_encoder_states
Reorder encoder states according to a new set of indices. This is an abstract method, and *must* be implemented by the user.
models/transformer.py:1209
Methodtrain
(self)
run.py:416
← previous101–130 of 130, ranked by callers