projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: a84e873) | patch
sparse.mm(S, D) (#14526)
authorWei Yang <weiyang@fb.com>
Fri, 30 Nov 2018 22:08:35 +0000 (14:08 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 30 Nov 2018 22:15:34 +0000 (14:15 -0800)
commitc3a2b1e15593b3b1fce331c5f6e212acee1c4594
treef177c98f4b10a590ed3d99be5ec4ffe4ef1bc294tree | snapshot
parenta84e873bb156080ea76ab182171b1f3b4d5395f6commit | diff
sparse.mm(S, D) (#14526)

Summary:
- add `sparse.mm(S, D)` with backward
- for `sparse.addmm()`, relax input constraint so that sparse matrix input doesn't have to coalesced
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14526

Reviewed By: ezyang

Differential Revision: D13252990

Pulled By: weiyangfb

fbshipit-source-id: 8fdb14144405a2122d4b8447ad4055cd0330e6e8
aten/src/ATen/native/native_functions.yaml diff | blob | history
aten/src/ATen/native/sparse/SparseTensorMath.cpp diff | blob | history
docs/source/sparse.rst diff | blob | history
test/test_sparse.py diff | blob | history
tools/autograd/templates/Functions.cpp diff | blob | history
torch/sparse/__init__.py diff | blob | history
Domain: Machine Learning / ML Framework;