两者都是三个输入参数,第一个是判断条件,第二个是符合条件的设置值,第三个是不满足条件的设置值。
区别在于,torch要将设置值全部改为Tensor类型的
mask_true = torch.where(torch.isnan(y_true), torch.full_like(y_true, 0), torch.full_like(y_true, 1))
mask_true = np.where(np.isnan(y_true), 0, 1)
两者都是三个输入参数,第一个是判断条件,第二个是符合条件的设置值,第三个是不满足条件的设置值。
区别在于,torch要将设置值全部改为Tensor类型的
mask_true = torch.where(torch.isnan(y_true), torch.full_like(y_true, 0), torch.full_like(y_true, 1))
mask_true = np.where(np.isnan(y_true), 0, 1)