Open fantasysponge opened 3 years ago
energy_new = torch.max(energy, -1, keepdim=True)[0].expand_as(energy)-energy what's meaning for this line?
energy_new = torch.max(energy, -1, keepdim=True)[0].expand_as(energy)-energy what's meaning for this line?