I was looking at torch.split() and apparently it splits a tensor into chunks along a given dimension, so if I do a = torch.tensor([1,2]) ...
確定! 回上一頁