Skip to content

维度问题 #1

@FYYFU

Description

@FYYFU

在utils.py 中 scaled dot-product attention实现函数中,返回的量的维度书写出现了错误,torch.matmul(p_attn, value) 的维度应该为 (batch_size, h, max_len, max_len)

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