losses = torch.where(
(w>absolute_x), #条件
w * torch.log(1.0 + absolute_x / epsilon), #a张量
absolute_x - c #b张量
)
torch.where实现了条件下多维a,b两段补充
losses = torch.where(
(w>absolute_x), #条件
w * torch.log(1.0 + absolute_x / epsilon), #a张量
absolute_x - c #b张量
)
torch.where实现了条件下多维a,b两段补充