def test_LogSoftmax(tmpdir, dtype): with C.default_options(dtype=dtype): model = C.log_softmax(np.array([[1, 1, 2, 3]]).astype(dtype)) verify_no_input(model, tmpdir, 'LogSoftmax_0')
def test_LogSoftmax(tmpdir): model = C.log_softmax([[1, 1, 2, 3]]) verify_no_input(model, tmpdir, 'LogSoftmax_0')
def test_LogSoftmax(tmpdir, dtype): with C.default_options(dtype = dtype): model = C.log_softmax(np.array([[1, 1, 2, 3]]).astype(dtype)) verify_no_input(model, tmpdir, 'LogSoftmax_0')