diffusion_rs_common::nn::ops

Function swiglu

Source
pub fn swiglu(xs: &Tensor) -> Result<Tensor>