Python stop_ifの例

コード例 #1

def _test_loop():
    x = ops.Const(5)
    y = ops.Const(0)
    with ops.loop():
        ops.stop_if(ops.EQ([x, ops.Const(0)]))
        ops.Add([x, ops.Const(-1)], [x])
        ops.Add([y, ops.Const(1)], [y])
    return y

コード例 #2

ファイル: net_builder_test.py プロジェクト: Ralfhund/caffe2

def _test_loop():
    x = ops.Const(5)
    y = ops.Const(0)
    with ops.loop():
        ops.stop_if(ops.EQ([x, ops.Const(0)]))
        ops.Add([x, ops.Const(-1)], [x])
        ops.Add([y, ops.Const(1)], [y])
    return y

コード例 #3

def _test_inner_stop(x):
    ops.stop_if(ops.LT([x, ops.Const(5)]))

コード例 #4

ファイル: net_builder_test.py プロジェクト: Ralfhund/caffe2

def _test_inner_stop(x):
    ops.stop_if(ops.LT([x, ops.Const(5)]))