diff --git a/python/paddle/v2/__init__.py b/python/paddle/v2/__init__.py index c0a2bdc4259c3f48e8e7fdf4c8d8b2584253875a..cf01f37a3324ff0ea5dfb0ba588c6187a57f4666 100644 --- a/python/paddle/v2/__init__.py +++ b/python/paddle/v2/__init__.py @@ -18,11 +18,12 @@ import parameters import trainer import event import data_type +import topology import py_paddle.swig_paddle as api __all__ = [ 'optimizer', 'layer', 'activation', 'parameters', 'init', 'trainer', - 'event', 'data_type.py' + 'event', 'data_type', 'topology' ] diff --git a/python/paddle/v2/topology.py b/python/paddle/v2/topology.py new file mode 100644 index 0000000000000000000000000000000000000000..ddba1b2345e5eb7d27b4943fd6ec834533ccf6a9 --- /dev/null +++ b/python/paddle/v2/topology.py @@ -0,0 +1,44 @@ +# Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserved +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +from . import layer + +__all__ = ['Topology'] + + +class Topology(object): + """ + Topology is used to store the information about all layers + and network configs. + """ + + def __init__(self, cost): + self.cost = cost + self.__model_config__ = layer.parse_network(cost) + + def __call__(self): + return self.__model_config__ + + def get_layer(self, name): + """ + get layer by layer name + :param name: + :return: + """ + pass + + def data_type(self): + """ + """ + pass