dmlc--dgl
xnouhz
0b47e86803
* [example] arma * update * update * update * update * update * [example] dimenet * [docs] update dimenet * [docs] update tf results * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update Co-authored-by: Mufei Li <mufeili1996@gmail.com>
8 行
181 B
Python
8 行
181 B
Python
import torch
|
|
|
|
def swish(x):
|
|
"""
|
|
Swish activation function,
|
|
from Ramachandran, Zopf, Le 2017. "Searching for Activation Functions"
|
|
"""
|
|
return x * torch.sigmoid(x) |