NNOIR的API
nnoir的Python项目详细描述
nnoir
安装
pip install nnoir
示例
创建并保存
inputs = [nnoir.Value(b'v0', dtype='<f4', shape=(10,10)),
nnoir.Value(b'v1', dtype='<f4', shape=(10,10))]
outputs = [nnoir.Value(b'v2', dtype='<f4', shape=(10,10))]
nodes = inputs + outputs
input_names = [ x.name for x in inputs ]
output_names = [ x.name for x in outputs ]
functions = [nnoir.functions.Add(input_names, output_names)]
result = nnoir.NNOIR(b'Add', b'add_test', '0.1', input_names, output_names, nodes, functions)
result.dump('add.nnoir')
负荷
add_nnoir = nnoir.load('add.nnoir')