• Jesse Grabowski's avatar
    Add `linalg.block_diag` and sparse equivalent (#576) · c4ae6e34
    Jesse Grabowski 提交于
    * Copy `block_diag` and support functions from `pymc.math`
    
    * Evaluate output in sphinx code example
    Co-authored-by: 's avatarRicardo Vieira <28983449+ricardoV94@users.noreply.github.com>
    
    * Test type equivalence with `isinstance` instead of `==`
    Co-authored-by: 's avatarRicardo Vieira <28983449+ricardoV94@users.noreply.github.com>
    
    * Typo in test function
    
    * Split `block_diag` into sparse and dense version
    
    Closely follow scipy function signature for `block_diag`
    
    * Use `as_sparse_or_tensor_variable` in `SparseBlockDiagonalMatrix` to allow sparse matrix inputs to `pytensor.sparse.block_diag`
    
    * Test sparse and dense inputs to `pytensor.sparse.block_diag`
    
    * Add numba overload for `pytensor.tensor.slinalg.block_diag`
    
    * add jax overload for `pytensor.tensor.slinalg.block_diag`
    
    * Move stand-alone `block_diag_grad` function into `grad` method
    
    * Add `format` prop to `SparseBlockDiagonalMatrix`
    
    * Use `compare_numba_and_py` in `numba\test_slinalg.py::test_block_diag`
    
    * Add support for Blockwise to `slinalg.block_diag`
    
    * Add gradient test
    
    Remove `Matrix` from `BlockDiagonal` and `SparseBlockDiagonal` `Op` names
    
    Correct errors in docstrings
    
    Move input validation to a shared class method
    
    * Remove `gufunc_signature` from `__props__`
    Co-authored-by: 's avatarRicardo Vieira <28983449+ricardoV94@users.noreply.github.com>
    
    * Implement correct `__props__` for subclasses of `BaseBlockMatrix`
    
    ---------
    Co-authored-by: 's avatarRicardo Vieira <28983449+ricardoV94@users.noreply.github.com>
    c4ae6e34
名称
最后提交
最后更新
.github 正在载入提交数据...
bin 正在载入提交数据...
doc 正在载入提交数据...
pytensor 正在载入提交数据...
scripts 正在载入提交数据...
tests 正在载入提交数据...
.gitattributes 正在载入提交数据...
.gitignore 正在载入提交数据...
.pre-commit-config.yaml 正在载入提交数据...
CODE_OF_CONDUCT.md 正在载入提交数据...
CONTRIBUTING.md 正在载入提交数据...
GOVERNANCE.md 正在载入提交数据...
LICENSE.txt 正在载入提交数据...
MANIFEST.in 正在载入提交数据...
README.rst 正在载入提交数据...
codecov.yml 正在载入提交数据...
conftest.py 正在载入提交数据...
environment.yml 正在载入提交数据...
pyproject.toml 正在载入提交数据...
readthedocs.yml 正在载入提交数据...
setup.cfg 正在载入提交数据...
setup.py 正在载入提交数据...
versioneer.py 正在载入提交数据...