Shortcuts

torch.index_add

torch.index_add(input, dim, index, source, *, alpha=1, out=None) 张量

参见 index_add_() 函数描述。

优云智算