torch.special

若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.xlogy |
是 |
支持fp16,fp32,uint8,int8,int16,int32,int64,bool |
父主题: PyTorch原生API支持度