def _squeeze(operand, *, dimensions, _in_avals, _out_aval):
    op_shape = _in_avals[0].shape
    new_shape = tuple((d for (i, d) in enumerate(op_shape) if (i not in dimensions)))
    new_shape_tf = _eval_shape(new_shape)
    return tf.reshape(operand, new_shape_tf)
def _squeeze(operand, *, dimensions, _in_avals, _out_aval):
    op_shape = _in_avals[0].shape
    new_shape = tuple((d for (i, d) in enumerate(op_shape) if (i not in dimensions)))
    new_shape_tf = _eval_shape(new_shape)
    return tf.reshape(operand, new_shape_tf)