frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add torch.nn.GELU for GELU activation (#28944)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/28944

Add torch.nn.GELU for GELU activation

Test Plan: buck test mode/dev-nosan //caffe2/test:nn -- "GELU"

Reviewed By: hl475, houseroad

Differential Revision: D18240946

fbshipit-source-id: 6284b30def9bd4c12bf7fb2ed08b1b2f0310bb78
This commit is contained in:
Xiaomeng Yang
2019-11-03 21:53:32 -08:00
committed by Facebook Github Bot
parent 3bffb730b6
commit 2460dced8f
23 changed files with 304 additions and 190 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docs/source/scripts/build_activation_images.py
View File

@ -37,6 +37,7 @@ functions = [
'RReLU',
'SELU',
'CELU',
'GELU',
'Sigmoid',
'Softplus',
'Softshrink',