Creating a random tensor x = torch.randn(2, 3) # normal distribution (rand(2,3) -> uniform ... LongTensor([0, 2]) chosen = torch.index_select(x, dim=1, ...
確定! 回上一頁