torch.special

API名称

是否支持

限制与说明

torch.special.logsumexp

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

torch.special.xlogy

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