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