Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/keras/src/layers/core/identity.py: 88%

8 statements  

« prev     ^ index     » next       coverage.py v7.4.0, created at 2024-01-03 07:57 +0000

1# Copyright 2022 The TensorFlow Authors. 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"""Contains the Identity layer.""" 

16 

17import tensorflow.compat.v2 as tf 

18 

19from keras.src.engine.base_layer import Layer 

20 

21# isort: off 

22from tensorflow.python.util.tf_export import keras_export 

23 

24 

25@keras_export("keras.layers.Identity") 

26class Identity(Layer): 

27 """Identity layer. 

28 

29 This layer should be used as a placeholder when no operation is to be 

30 performed. The layer is argument insensitive, and returns its `inputs` 

31 argument as output. 

32 

33 Args: 

34 name: Optional name for the layer instance. 

35 """ 

36 

37 def call(self, inputs): 

38 return tf.nest.map_structure(tf.identity, inputs) 

39