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

支持fp32,fp16,int64,bool

torch.special.erfc

支持fp32,fp16,int64,bool

torch.special.erfcx

     

torch.special.erfinv

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

torch.special.exp2

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

torch.special.expit

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

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

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

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