torch.special

若API未标明支持情况,则代表该API的支持情况待验证。

API名称

是否支持

限制与说明

torch.special.airy_ai

  

torch.special.bessel_j0

  

torch.special.bessel_j1

  

torch.special.digamma

     

torch.special.entr

     

torch.special.erf

     

torch.special.erfc

     

torch.special.erfcx

     

torch.special.erfinv

     

torch.special.exp2

     

torch.special.expit

     

torch.special.expm1

     

torch.special.gammainc

     

torch.special.gammaincc

     

torch.special.gammaln

     

torch.special.i0

     

torch.special.i0e

     

torch.special.i1

     

torch.special.i1e

     

torch.special.log1p

     

torch.special.log_ndtr

  

torch.special.log_softmax

     

torch.special.logit

     

torch.special.logsumexp

支持fp32,fp16,uint8,int8,int16,int32,int64,bool

torch.special.multigammaln

     

torch.special.ndtr

     

torch.special.ndtri

     

torch.special.polygamma

     

torch.special.psi

     

torch.special.round

     

torch.special.scaled_modified_bessel_k0

  

torch.special.scaled_modified_bessel_k1

  

torch.special.sinc

     

torch.special.softmax

     

torch.special.spherical_bessel_j0

  

torch.special.xlog1py

     

torch.special.xlogy

支持fp16,fp32,int8,uint8,int16,int32,int64,bool

torch.special.zeta