Skip to content

麻烦问下这里是什么意思,谢谢! #2

@alwayslikethat

Description

@alwayslikethat
    for i in range(BATCH_SIZE):
        s_len = seqlen_1d[i]
        mask[i, 0:s_len] = 1
    mask = torch.FloatTensor(mask).to(self.device)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions