Skip to content

MultiHeadAttention op shall return attention probabilities #23124

@amancini-N

Description

@amancini-N

Support pointer-generator networks in MultiHeadAttention op. Specifically, MultiHeadAttention op shall have an additional output returning the attention probabilities (softmax result) out.

Metadata

Metadata

Assignees

No one assigned

    Labels

    core runtimeissues related to core runtimestaleissues that have not been addressed in a while; categorized by a bot

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions