tests/tensor/test_special.py · 9d7c2ec9c4fbf8d61cb2abc2ca820cc0cb35cdf9 · testgroup / pytensor · GitLab

Blame History Permalink

Implement vectorize_node for Softmax and Argmax Ops · 65967fe2
由 Ricardo Vieira 提交于 1月 05, 2024
```
Also refactors shared logic for other batch axed Ops
```
65967fe2

test_special.py 6.7 KB