• Harshvir Sandhu's avatar
    Add docs on implementing PyTorch Ops (and CumOp) (#837) · 781073b6
    Harshvir Sandhu 提交于
    * Add Pytorch support for Cum Op
    
    * Modify test for Cum op
    
    * Raise TypeError if axis not int or None
    
    * Fix init method of CumOp
    
    * Extend tutorial on documentation for Pytorch
    
    * Add tab for Numba
    
    * Add intersphinx mapping
    
    * Parametrize dtype
    781073b6
conf.py 9.2 KB