index_put

input = torch.randn(2,2)

part = torch.tensor([[1, 2]])

ind0 = torch.arange(0,1)
ind1 = torch.arange(0,2)

input = torch.index_put(input, (ind0, ind1), part)


---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-39-8e225bbb2c1f> in <module>()
      6 ind1 = torch.arange(0,2)
      7
----> 8 input = torch.index_put(input, (ind0, ind1), part)

RuntimeError: expand(torch.LongTensor{[1, 2]}, size=[2]): the number of sizes provided (1) must be greater or equal to the number of dimensions in the tensor (2)