o
    wZŽhÚ  ã                   @   s    d dl mZmZmZ d dlmZmZm	Z	m
Z
mZ d dlmZmZ d dlmZmZmZmZ d dlmZ d dlmZmZ d dlmZmZ d dlmZmZ g d	¢Zd
S )é   )ÚparametrizationsÚrnnÚ	stateless)Ú_clip_grads_with_norm_Ú_get_total_normÚclip_grad_normÚclip_grad_norm_Úclip_grad_value_)Úparameters_to_vectorÚvector_to_parameters)Úfuse_conv_bn_evalÚfuse_conv_bn_weightsÚfuse_linear_bn_evalÚfuse_linear_bn_weights)Ú	skip_init)Ú#convert_conv2d_weight_memory_formatÚ#convert_conv3d_weight_memory_format)Úremove_spectral_normÚspectral_norm)Úremove_weight_normÚweight_norm)r   r   Úclip_grads_with_norm_r	   r   r   r   r   r   r   Úget_total_normr
   r   r   r   r   r   r   r   r   r   N)Ú r   r   r   Z	clip_gradr   r   r   r   r   r   r	   Zconvert_parametersr
   r   Zfusionr   r   r   r   Úinitr   Zmemory_formatr   r   r   r   r   r   Ú__all__© r   r   úF/var/www/auris/lib/python3.10/site-packages/torch/nn/utils/__init__.pyÚ<module>   s    