torch_build_nn
