Tensor Views

若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