rust-bert 0.7.2

Ready-to-use NLP pipelines and transformer-based models (BERT, DistilBERT, GPT2,...)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14

use tch::Tensor;
use std::f64::consts::PI;

pub fn _gelu(x: &Tensor) -> Tensor { x * 0.5 * (1.0 + (x / ((2.0 as f64).sqrt())).erf()) }

pub fn _relu(x: &Tensor) -> Tensor { x.relu() }

pub fn _swish(x: &Tensor) -> Tensor { x * x.sigmoid() }

pub fn _mish(x: &Tensor) -> Tensor { x * (x.softplus().tanh()) }

pub fn _gelu_new(x: &Tensor) -> Tensor { x * 0.5 * (((x.pow(3.0f64) * 0.044715 + x) * ((2f64 / PI).sqrt())).tanh() + 1) }

pub fn _tanh(x: &Tensor) -> Tensor { x.tanh() }