Exemplos de sanitize_dynamic_axes em Python

Exemplo n.º 1

Arquivo: utils_test.py Projeto: OlegBoulanov/CNTK

def test_axes():
    axes = [C.Axis.default_batch_axis(), C.Axis.default_dynamic_axis()]
    assert tuple(axes) == C.Axis.default_input_variable_dynamic_axes()
    assert sanitize_dynamic_axes(axes) == \
            tuple(reversed(C.Axis.default_input_variable_dynamic_axes()))

    assert (C.Axis.default_dynamic_axis(),) == \
            sanitize_dynamic_axes(C.Axis.default_dynamic_axis())

Exemplo n.º 2

Arquivo: utils_test.py Projeto: ykv001/CNTK

def test_axes():
    axes = [C.Axis.default_batch_axis(), C.Axis.default_dynamic_axis()]
    assert tuple(axes) == C.Axis.default_input_variable_dynamic_axes()
    assert sanitize_dynamic_axes(axes) == \
            tuple(reversed(C.Axis.default_input_variable_dynamic_axes()))

    assert (C.Axis.default_dynamic_axis(),) == \
            sanitize_dynamic_axes(C.Axis.default_dynamic_axis())