python 命令行传参

def parse_args():
    import argparse
    parser = argparse.ArgumentParser(description='tensor analysis')

    parser.add_argument('-i',
                        '--input',
                        default='.',
                        help='path of your datas')
    parser.add_argument('-n',
                        '--number',
                        default=50,
                        type=int,
                        help='number of result you want to see')
    parser.add_argument('-o',
                        '--output',
                        default='.',
                        help='path where to save')

    args = parser.parse_args()

    return args


def main(args):
    is_bfloat = IsBf16()

    maybe_print("is_bfloat {}".format(is_bfloat))

    in_f0 = open(args.input, "rb")
    in_c0 = in_f0.read()

if __name__ == "__main__":
    args = parse_args()
    main(args)


$ python xxxx.py --input xxxx