Function torch::nn::functional::multi_head_attention_forward

• Defined in File activation.h

Function Documentation

inline std::tuple<Tensor, Tensor> torch::nn::functional::multi_head_attention_forward(const Tensor &query, const Tensor &key, const Tensor &value, const MultiheadAttentionForwardFuncOptions &options)