本文档由 AI 生成。如果您发现任何错误或有改进建议,欢迎贡献! 在 GitHub 上编辑UNetSelfAttentionMultiply 节点对 UNet 模型中的自注意力机制的查询、键、值和输出分量应用乘法因子。它允许您缩放注意力计算的不同部分,以试验注意力权重如何影响模型的行为。
输入
| 参数 | 数据类型 | 是否必填 | 范围 | 描述 |
|---|---|---|---|---|
模型 | MODEL | 是 | - | 要使用注意力缩放因子进行修改的 UNet 模型 |
q | FLOAT | 否 | 0.0 - 10.0 | 查询分量的乘法因子(默认值:1.0) |
k | FLOAT | 否 | 0.0 - 10.0 | 键分量的乘法因子(默认值:1.0) |
v | FLOAT | 否 | 0.0 - 10.0 | 值分量的乘法因子(默认值:1.0) |
输出 | FLOAT | 否 | 0.0 - 10.0 | 输出分量的乘法因子(默认值:1.0) |
输出
| 输出名称 | 数据类型 | 描述 |
|---|---|---|
MODEL | MODEL | 经过注意力分量缩放后的修改版 UNet 模型 |
Source fingerprint (SHA-256):
ee6328c6cba44d30d2e219a2af04bb3d3d9adeaabb959a46f87b3b299dfe2f43