若API未标明支持情况，则代表该API的支持情况待验证。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.adjoint |
是 |
可通过CPU实现 |
torch.as_strided |
是 |
支持fp32 |
torch.detach |
是 |
|
torch.diagonal |
是 |
支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128 |
torch.expand |
是 |
|
torch.expand_as |
是 |
|
torch.movedim |
是 |
支持bf16，fp16，fp32，fp64，uint8，int8，int16，int32，int64，bool |
torch.narrow |
是 |
支持bf16，fp16，fp32，fp64，int8，uint8，int16，int32，int64，bool |
torch.permute |
是 |
支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128 |
torch.select |
||
torch.squeeze |
是 |
支持bf16，fp16，fp32，fp64，uint8，int8，int16，int32，int64，bool |
torch.transpose |
是 |
支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128 |
torch.t |
是 |
支持bf16，fp16，fp32，fp64，int8，uint8，int16，int32，int64，bool |
torch.T |
是 |
支持bf16，fp16，fp32，fp64，int8，uint8，int16，int32，int64，bool |
torch.H |
是 |
|
torch.mT |
是 |
|
torch.mH |
是 |
|
torch.real |
是 |
支持fp32，fp16，complex64，complex128 |
torch.imag |
否 |
|
torch.view_as_real |
是 |
支持complex64 |
torch.unflatten |
是 |
|
torch.unfold |
是 |
|
torch.unsqueeze |
是 |
支持bf16，fp16，fp32，fp64，int8，uint8，int16，int32，int64，bool |
torch.view |
||
torch.view_as |
是 |
|
torch.unbind |
||
torch.split |
是 |
支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128 |
torch.hsplit |
是 |
支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128 |
torch.vsplit |
是 |
支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128 |
torch.tensor_split |
是 |
支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128 |
torch.split_with_sizes |
是 |
|
torch.swapaxes |
是 |
支持bf16，fp16，fp32，fp64，uint8，int8，int16，int32，int64，bool |
torch.swapdims |
是 |
支持bf16，fp16，fp32，fp64，uint8，int8，int16，int32，int64，bool |
torch.chunk |
是 |
支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128 |
torch.indices |
||
torch.values |