zurutech / ashpy
1
# Copyright 2019 Zuru Tech HK Limited. All Rights Reserved.
2
#
3
# Licensed under the Apache License, Version 2.0 (the "License");
4
# you may not use this file except in compliance with the License.
5
# You may obtain a copy of the License at
6
#
7
# http://www.apache.org/licenses/LICENSE-2.0
8
#
9
# Unless required by applicable law or agreed to in writing, software
10
# distributed under the License is distributed on an "AS IS" BASIS,
11
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
# See the License for the specific language governing permissions and
13
# limitations under the License.
14

15 2
"""
16
Collection of layers.
17

18
.. currentmodule:: ashpy.layers
19

20
.. rubric:: Layers
21

22
.. autosummary::
23
    :nosignatures:
24
    :toctree: layers
25

26
    instance_normalization.InstanceNormalization
27
    attention.Attention
28

29
"""
30

31 2
from ashpy.layers.attention import Attention
32 2
from ashpy.layers.instance_normalization import InstanceNormalization
33

34 2
__ALL__ = ["Attention", "InstanceNormalization"]

Read our documentation on viewing source code .

Loading