pytorch
a8ef4d3f - Provide 'out' parameter for 'tensordot' (#47278)

Commit
4 years ago
Provide 'out' parameter for 'tensordot' (#47278) Summary: Fixes https://github.com/pytorch/pytorch/issues/42102 Added an optional out parameter to the tensordot operation to allow using buffers. Pull Request resolved: https://github.com/pytorch/pytorch/pull/47278 Test Plan: pytest test/test_torch.py -k tensordot -v Reviewed By: agolynski Differential Revision: D24706258 Pulled By: H-Huang fbshipit-source-id: eb4bcd114795f67de3a670291034107d2826ea69
Author
Parents
Loading