跳转到主要内容
本文档由 AI 生成。如果您发现任何错误或有改进建议,欢迎贡献! 在 GitHub 上编辑
UNetSelfAttentionMultiply 节点对 UNet 模型中的自注意力机制的查询、键、值和输出分量应用乘法因子。它允许您缩放注意力计算的不同部分,以试验注意力权重如何影响模型的行为。

输入

参数数据类型是否必填范围描述
模型MODEL-要使用注意力缩放因子进行修改的 UNet 模型
qFLOAT0.0 - 10.0查询分量的乘法因子(默认值:1.0)
kFLOAT0.0 - 10.0键分量的乘法因子(默认值:1.0)
vFLOAT0.0 - 10.0值分量的乘法因子(默认值:1.0)
输出FLOAT0.0 - 10.0输出分量的乘法因子(默认值:1.0)

输出

输出名称数据类型描述
MODELMODEL经过注意力分量缩放后的修改版 UNet 模型

Source fingerprint (SHA-256): ee6328c6cba44d30d2e219a2af04bb3d3d9adeaabb959a46f87b3b299dfe2f43