Skip to content

Conversation

@chengtyao2-design
Copy link

Description

本 PR 在 InfiniCore 中新增了 Softmax 算子的 Python 接口及测试用例。

Implemented Features

  • Softmax Op: 实现了 Softmax 算子的 Python 绑定。
    • python/infinicore/ops/softmax.py
  • Tests: 添加了 Softmax 算子的测试脚本。
    • test/infinicore/ops/softmax.py
    • testop.py

Testing

  • 经本地测试,算子可以正常调用并计算正确结果。

@chengtyao2-design chengtyao2-design requested a review from a team January 14, 2026 09:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants