Searched defs:generate_square_subsequent_mask (Results 1 – 2 of 2) sorted by relevance
465 Tensor TransformerImpl::generate_square_subsequent_mask(int64_t sz) { in generate_square_subsequent_mask() function in torch::nn::TransformerImpl
291 def generate_square_subsequent_mask( member in Transformer