跳转到主要内容
本文档由 AI 生成。如果您发现任何错误或有改进建议,欢迎贡献! 在 GitHub 上编辑
UNetCrossAttentionMultiply 节点用于对 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)

输出结果

输出名称数据类型描述
模型MODEL经过交叉注意力组件缩放处理后的 UNet 模型