pytorch 报错 RuntimeError: Invalid index in scatter at

很大可能是因为index的值超出了范围，
比如

import torch

batch_size = 2

hidden_dim = 5

x = torch.zeros(batch_size, hidden_dim).scatter_(dim=-1,
                               index=torch.LongTensor([[2],[1]]),
                               value=2)

index如果传入​​torch.LongTensor([[6],[1]])​​的时候