• 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
名称
最后提交
最后更新
..
conv 正在载入提交数据...
random 正在载入提交数据...
rewriting 正在载入提交数据...
__init__.py 正在载入提交数据...
test_basic.py 正在载入提交数据...
test_blas.py 正在载入提交数据...
test_blas_c.py 正在载入提交数据...
test_blas_scipy.py 正在载入提交数据...
test_blockwise.py 正在载入提交数据...
test_casting.py 正在载入提交数据...
test_complex.py 正在载入提交数据...
test_elemwise.py 正在载入提交数据...
test_extra_ops.py 正在载入提交数据...
test_fft.py 正在载入提交数据...
test_fourier.py 正在载入提交数据...
test_functional.py 正在载入提交数据...
test_gc.py 正在载入提交数据...
test_inplace.py 正在载入提交数据...
test_io.py 正在载入提交数据...
test_keepdims.py 正在载入提交数据...
test_math.py 正在载入提交数据...
test_math_scipy.py 正在载入提交数据...
test_merge.py 正在载入提交数据...
test_nlinalg.py 正在载入提交数据...
test_shape.py 正在载入提交数据...
test_sharedvar.py 正在载入提交数据...
test_slinalg.py 正在载入提交数据...
test_sort.py 正在载入提交数据...
test_special.py 正在载入提交数据...
test_subtensor.py 正在载入提交数据...
test_type.py 正在载入提交数据...
test_type_other.py 正在载入提交数据...
test_utils.py 正在载入提交数据...
test_variable.py 正在载入提交数据...
test_xlogx.py 正在载入提交数据...
utils.py 正在载入提交数据...