pytorch
71be5fe5 - add support for {ones,zeros,full,rand,randn}_like ops (#28981)

Commit
5 years ago
add support for {ones,zeros,full,rand,randn}_like ops (#28981) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/28981 This PR adds support for calling those functions on named tensors. The implementation is not the nicest: in the future we have plans to merge names into TensorOptions, at which point we don't need the extra branches that check if the tensor has names. Right now, however, these functions are very useful to have (in particular, ones_like is used by autograd to generate gradients). Test Plan: - Added tests for each of these Differential Revision: D18270937 Pulled By: zou3519 fbshipit-source-id: 720739ff0474449a960b81728345a4250becbfc3
Author
Parents
Loading