Skip to content

P
pytensor
切换分支/标签
查找文件
BlameHistoryPermalink
  • Dan F-M's avatar
    adding JAX conversion for BatchedDot · abde1a16
    由 Dan F-M 提交于 
    BatchedDot only supports tensor3

fixing broadcasting behavior in BatchedDot

adding test for TypeError on dimension mismatch

removing extra tests
    abde1a16
test_jax.py 30.9 KB