若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 |