torch.special

Note

若API“是否支持”为“是”,“限制与说明”为“-”,说明此API和原生API支持度保持一致。

API名称 是否支持 限制与说明
torch.special.erf 支持fp16,fp32,int64,bool
torch.special.erfc 支持fp16,fp32,int64,bool
torch.special.erfinv 支持fp16,fp32,uint8,int8,int16,int32,int64,bool
torch.special.exp2 支持fp16,fp32,uint8,int8,int16,int32,int64,bool
torch.special.expit 支持fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128
torch.special.log1p 支持fp16,fp32,uint8,int8,int16,int32,int64,bool
torch.special.logsumexp 支持fp16,fp32,uint8,int8,int16,int32,int64,bool
torch.special.ndtr -
torch.special.xlogy 支持fp16,fp32,uint8,int8,int16,int32,int64,bool