pytorch
3c17cbb6 - fix #30480 torch.normal shape checking is broken (#32243)

Commit
4 years ago
fix #30480 torch.normal shape checking is broken (#32243) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/32243 Following what gchanan proposed in #30480 - If the (logical) shapes of mean and std are broadcastable, we broadcast them for the output Done in tensor iterator already. - If the (logical) shapes of mean and std are not broadcastable and they have the same number of elements, we fall back to the old behavior (pick the shape of mean) Done by reshape std to the same shape of mean. - If the (logical) shapes of mean and std are not broadcastable and don't have the same number of elements, we error out. Done by tensor iterator already. Test Plan: Imported from OSS Differential Revision: D19417087 Pulled By: glaringlee fbshipit-source-id: 1c4bc7df923110a803620b9e2abd11a7151fc33e
Author
lixinyu
Parents
Loading