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)]))