pain_gain0的博客

原创 output = torch.Tensor.scatter_(dim, index, src)

dim = 0, 按照数值方向操作； dim = 1, 按照水平方向操作； a = torch.rand(2, 5) print(a) b = torch.zeros(3, 5).scatter_(0, torch.tensor([[0, 1, 2, 0, 0], [2, 0, 0, 1, 2]]), a) print(b) output = torch.Tensor.scatter_(...