Modifying the agent attention to accept NHWC tensor

[alvinwong64]

Hi i would like to plug in the agentattention module to test for the performance of it on a yolo-segmentation model, how could i modify it to accept input in the form of NHWC, thanks!

[2406583758qq]

嗨，我想插入 agentattention 模块来测试它在 yolo-segmentation 模型上的性能，我该如何修改它以接受 NHWC 形式的输入，谢谢！

Have you solved this problem?