test_objectives.py 文件源码

def test_SoftmaxCategoricalCrossEntropy():
    from npdl.objectives import SoftmaxCategoricalCrossEntropy

    obj = SoftmaxCategoricalCrossEntropy()

    outputs = np.random.random((10, 20))
    targets = np.random.random((10, 20))

    f_res = obj.forward(outputs, targets)
    b_res = obj.backward(outputs, targets)

    assert np.ndim(f_res) == 0
    assert np.ndim(b_res) == 2