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 |
父主题： PyTorch原生API
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.special.logsumexp |
是 |
支持fp16，fp32，uint8，int8，int16，int32，int64，bool |
torch.special.xlogy |
是 |
支持fp16，fp32，uint8，int8，int16，int32，int64，bool |