圣托里尼的日落啊~的博客

torch.repeatimport torchv_sum = torch.tensor([[1,3,0],[1,0,0]])print(v_sum)mask_index = torch.nonzero(v_sum == 0)print(mask_index)q = torch.rand([2,1,6])print(q)q_expand = q.repeat(1,3,1)...

scatter() 和 scatter_() 的作用是一样的，只不过 scatter() 不会直接修改原来的 Tensor，而 scatter_() 会PyTorch 中，一般函数加下划线代表直接在原来的 Tensor 上修改scatter(dim, index, src) 的参数有 3 个dim：沿着哪个维度进行索引index：用来 scatter 的元素索引src：用来 scatt...

import torchimport torch.nn.functional as Fpred = torch.tensor([[-0.4089,-1.2471,0.5907],[-0.4897,-0.8267,-0.7349],[0.5241,-0.1246,-0.4757]]).cuda()print(pred)'''tensor([[-0.4089, -1.2471, 0.59...