






'''simple docstring'''






from typing import Optional, Union

import torch
from torch import nn

from ...configuration_utils import ConfigMixin, register_to_config
from ...models.modeling_utils import ModelMixin




class a_    (    UpperCamelCase_		,      UpperCamelCase_							):







			@register_to_config
			def __init__(self					,   __a = 7_6_8					,   )			->      Any:
						"""simple docstring"""






						super().__init__()

						__snake_case				:      List[Any]     							= nn.Parameter(torch.zeros(1					,   __a))
						__snake_case				:      Dict     							= nn.Parameter(torch.ones(1					,   __a))







			def     SCREAMING_SNAKE_CASE__					(self					,   __a = None					,   __a = None					,   )			->      Dict:
						"""simple docstring"""






						__snake_case				:      Dict     							= nn.Parameter(self.mean.to(__a).to(__a))
						__snake_case				:      str     							= nn.Parameter(self.std.to(__a).to(__a))
						return self







			def     SCREAMING_SNAKE_CASE__					(self					,   __a)			->      List[Any]:
						"""simple docstring"""






						__snake_case				:      int     							= (embeds - self.mean) * 1.0 / self.std
						return embeds







			def     SCREAMING_SNAKE_CASE__					(self					,   __a)			->      List[str]:
						"""simple docstring"""






						__snake_case				:      Any     							= (embeds * self.std) + self.mean
						return embeds







'''simple docstring'''






# Copyright 2023 The HuggingFace Team. 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 typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


__A    	=   {'''configuration_timm_backbone''': ['''TimmBackboneConfig''']}


try:
      if not is_torch_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      __A    	=   ['''TimmBackbone''']


if TYPE_CHECKING:
      from .configuration_timm_backbone import TimmBackboneConfig

      try:
            if not is_torch_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .modeling_timm_backbone import TimmBackbone

else:
      import sys

      __A    	=   _LazyModule(__name__, globals()['''__file__'''], _import_structure, module_spec=__spec__)