Opened this issue 7 days ago · 0 comments
是否可以支持Softmax 3d 的算子, 很多vit的模型,包括ScaledDotProductAttention算子都需要