torch.slice works fine for fixed ranges (e.g. a[0:2]). If instead of particular number I use Var for Length (just for Length) (e.g. a[0, ...
確定! 回上一頁