CoCalc -- a.ipynb

GitHub Repository: Yiming-Wange/CS224n-2023-solution
Path: blob/main/a3/a.ipynb
³⁷⁶³ views

Kernel: base

In [1]:

import torch

In [7]:

q = torch.empty(5,5)
torch.index_select(q, 0, torch.tensor([1,3,3]))

Out[7]:

tensor([[0., 0., 0., 0., 0.],
        [0., 0., 0., 0., 0.],
        [0., 0., 0., 0., 0.]])