frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add determinant function on variable; Add backward on svd (#3816)

Browse Source 
* determinant on variable

* svd bwd
This commit is contained in:
Tongzhou Wang
2017-12-01 13:22:46 -05:00
committed by Soumith Chintala
parent 80c8635a7e
commit c681b03d37
20 changed files with 424 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tools/autograd/gen_python_functions.py
View File

@ -54,7 +54,9 @@ UNPACK_SELF = "auto& self_ = reinterpret_cast<THPVariable*>(self)->cdata;"
# to add an appropriate wrap() overload in torch/csrc/autograd/utils/wrap_outputs.h.
SUPPORTED_RETURN_TYPES = {
'Tensor', 'std::tuple<Tensor,Tensor>',
'std::tuple<Tensor,Tensor,Tensor>', 'std::vector<Tensor>',
'std::tuple<Tensor,Tensor,Tensor>',
'std::tuple<Tensor,Tensor,Tensor,Tensor>',
'std::vector<Tensor>',
'Scalar', 'bool', 'int64_t', 'void*'
}