mindformers.models.bloom.BloomLMHeadModel¶

class mindformers.models.bloom.BloomLMHeadModel(config=None)[源代码]¶

Provide bloom training loss or logits through network.

参数: config (BloomConfig) – The config of BloomModel.
返回: Tensor, the loss or logits of the network.

get_top_token_id(logits, current_index=None)[源代码]¶