pytorch
7828a220 - fix a bug in leakyReLU (#48265)

Commit
5 years ago
fix a bug in leakyReLU (#48265) Summary: The scale variable needs to be a scalar, otherwise it will report the following error: "RuntimeError: Cannot input a tensor of dimension other than 0 as a scalar argument" Pull Request resolved: https://github.com/pytorch/pytorch/pull/48265 Test Plan: Tested locally and the error disappeared. Reviewed By: zhizhengwu Differential Revision: D25105423 Pulled By: jerryzh168 fbshipit-source-id: 2a0df24cf7e40278a950bffe6e0a9552f99da1d1
Author
Zhi-Zheng Wu
Parents
Loading