Batched upper triangular, lower triangular (#15257)
authorvishwakftw <cs15btech11043@iith.ac.in>
Thu, 10 Jan 2019 03:36:20 +0000 (19:36 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 10 Jan 2019 03:46:39 +0000 (19:46 -0800)
commitb4c3268b23c30cb14b1a249e9566e0bd54c9bcd8
tree18ad9702c99049db938d6da585e2c92004a86198
parent5af9aaa5bbfbad51f420acde0f992279404032e5
Batched upper triangular, lower triangular (#15257)

Summary:
Changelog:

- Implements `triu` and `tril` for batches of 2D tensors.
- Remove TH/THC binding for `tril`
- Fix CUDA implementation
- Update docstrings for tril and triu.
- Remove mask-based `triu` and `tril` in cholesky forward and backward.
- Remove batched tril in torch.distributions.utils
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15257

Differential Revision: D13613888

Pulled By: mrshenli

fbshipit-source-id: 0949a05b9b8e974c1acfaf02a6284848ec5cc1c4
18 files changed:
aten/src/ATen/Declarations.cwrap
aten/src/ATen/native/BatchLinearAlgebra.cpp
aten/src/ATen/native/LegacyDefinitions.cpp
aten/src/ATen/native/LinearAlgebraUtils.h
aten/src/ATen/native/cuda/BatchLinearAlgebra.cu
aten/src/ATen/native/native_functions.yaml
aten/src/TH/generic/THTensorMath.h
aten/src/TH/generic/THTensorMoreMath.cpp
aten/src/THC/generic/THCTensorMath.h
aten/src/THC/generic/THCTensorMathPairwise.cu
test/common_methods_invocations.py
test/test_cuda.py
test/test_torch.py
tools/autograd/templates/Functions.cpp
torch/_torch_docs.py
torch/distributions/constraints.py
torch/distributions/utils.py
torch/functional.py