def test_BCELoss_reduce(): """ test_BCELoss """ loss = nn.BCELoss(reduction='mean') inputs_data = Tensor( np.array([[0.1, 0.2, 0.3], [0.5, 0.7, 0.9]]).astype(np.float32)) target_data = Tensor(np.array([[0, 1, 0], [0, 0, 1]]).astype(np.float32)) loss(inputs_data, target_data)
def test_BCELoss_weight(): """ test_BCELoss """ weight = Tensor( np.array([[1.0, 2.0, 3.0], [2.2, 2.6, 3.9]]).astype(np.float32)) loss = nn.BCELoss(weight=weight) inputs_data = Tensor( np.array([[0.1, 0.2, 0.3], [0.5, 0.7, 0.9]]).astype(np.float32)) target_data = Tensor(np.array([[0, 1, 0], [0, 0, 1]]).astype(np.float32)) loss(inputs_data, target_data)