若API未标明支持情况,则代表该API的支持情况待验证。
API名称 |
是否支持 |
限制与说明 |
---|---|---|
torch.adjoint |
是 |
|
torch.as_strided |
是 |
|
torch.detach |
是 |
|
torch.diagonal |
是 |
仅支持diagonal=0场景。 |
torch.expand |
是 |
|
torch.expand_as |
是 |
|
torch.movedim |
是 |
|
torch.narrow |
是 |
|
torch.permute |
是 |
|
torch.select |
是 |
|
torch.squeeze |
是 |
|
torch.transpose |
是 |
|
torch.t |
是 |
|
torch.T |
||
torch.H |
是 |
|
torch.mT |
是 |
|
torch.mH |
是 |
|
torch.real |
是 |
|
torch.imag |
否 |
|
torch.view_as_real |
否 |
|
torch.unflatten |
是 |
|
torch.unfold |
是 |
|
torch.unsqueeze |
是 |
|
torch.view |
||
torch.view_as |
是 |
|
torch.unbind |
是 |
|
torch.split |
是 |
|
torch.hsplit |
是 |
|
torch.vsplit |
是 |
只支持float16,float32,float64。 |
torch.tensor_split |
是 |
|
torch.split_with_sizes |
是 |
|
torch.swapaxes |
是 |
|
torch.swapdims |
是 |
|
torch.chunk |
是 |
|
torch.indices |
||
torch.values |