xrz000

Hi, I've seen the same problem. According to PyTorch's [documents](https://pytorch.org/docs/stable/tensors.html#torch.Tensor.index_add_), the `index` parameter should be a `LongTensor`. In `Nk.index_add_(0, argmax, Global.ones)`, `argmax` is an `IntTensor`, which causes the error. You...