Esempi in Python per squeeze

Esempio n. 1

File: to_relay.py Progetto: zheng-xq/tvm

def _squeeze(children, attrs, odtype='float32'):
    axis = None
    try:
        axis = [attrs.get_int('axis', None)]
    except ValueError:
        axis = axis or attrs.get_int_tuple('axis', None)

    return op.squeeze(children[0], axis)

Esempio n. 2

def _tvm_squeeze(inp, dim, name):
    return _op.squeeze(inp, axis=[dim])

Esempio n. 3

File: to_relay.py Progetto: hoseung2/Latency-Predictor

def _squeeze(children, attrs, odtype='float32'):
    axis = attrs.get_int_tuple('axis', None)
    axis = [axis] if isinstance(axis, int) else axis

    return op.squeeze(children[0], axis)