o % i4@sddlmZddlmZddlZddlmZmZm Z m Z m Z m Z ddl mZgdZddd dZd ddZd!d"ddZd!d#ddZd$ddZdS)%) annotations)OptionalN)ActDropoutNormPoolRelPosEmbedding split_args has_option)get_norm_layer get_act_layerget_dropout_layerget_pool_layername tuple | str spatial_dims int | NonechannelscCst|dkr tjSt|\}}t||f}t|}t|dr&d|vr&||d<t|dr3d|vr3||d<|di|S)a~ Create a normalization layer instance. For example, to create normalization layers: .. code-block:: python from monai.networks.layers import get_norm_layer g_layer = get_norm_layer(name=("group", {"num_groups": 1})) n_layer = get_norm_layer(name="instance", spatial_dims=2) Args: name: a normalization type string or a tuple of type string and parameters. spatial_dims: number of spatial dimensions of the input. channels: number of features/channels when the normalization layer requires this parameter but it is not specified in the norm parameters. num_features num_channelsN)torchnnIdentityr rdictr )rrr norm_name norm_args norm_typekw_argsrr]/home/dell461/cl/sdc2/last_ska_mid/HISourceFinder-master-l/src/monai/networks/layers/utils.pyr s   r cCs4|dkr tjSt|\}}t|}|di|S)a Create an activation layer instance. For example, to create activation layers: .. code-block:: python from monai.networks.layers import get_act_layer s_layer = get_act_layer(name="swish") p_layer = get_act_layer(name=("prelu", {"num_parameters": 1, "init": 0.25})) Args: name: an activation type string or a tuple of type string and parameters. rNr)rrrr r)rZact_nameZact_argsZact_typerrr"r 7s   r tuple | str | float | int dropout_dimcCsZ|dkr tjSt|ttfrtj}dt|i}nt|\}}t||f}|di|S)a Create a dropout layer instance. For example, to create dropout layers: .. code-block:: python from monai.networks.layers import get_dropout_layer d_layer = get_dropout_layer(name="dropout") a_layer = get_dropout_layer(name=("alphadropout", {"p": 0.25})) Args: name: a dropout ratio or a tuple of dropout type and parameters. dropout_dim: the spatial dimension of the dropout operation. rpNr) rrr isinstanceintfloatrZDROPOUTr )rr$Z drop_nameZ drop_argsZ drop_typerrr"rNs   rcCs8|dkr tjSt|\}}t||f}|di|S)a Create a pooling layer instance. For example, to create adaptiveavg layer: .. code-block:: python from monai.networks.layers import get_pool_layer pool_layer = get_pool_layer(("adaptiveavg", {"output_size": (1, 1, 1)}), spatial_dims=3) Args: name: a pooling type string or a tuple of type string and parameters. spatial_dims: number of spatial dimensions of the input. rNr)rrrr r)rrZ pool_nameZ pool_argsZ pool_typerrr"rks    r s_input_dimsOptional[tuple]c_dimr' num_headscsHt|\}}t||||d|}fdd|D}di|S)N)r)r+r,cs i|] \}}t|r||qSrr ).0kvZembedding_typerr" s z/get_rel_pos_embedding_layer..r)r ritems)rr)r+r,Zembedding_nameZembedding_argsr!rr0r"get_rel_pos_embedding_layers r3)rr)rrrrrr)rr)r)rr#r$r)rrrr)rrr)r*r+r'r,r') __future__rtypingrtorch.nnrmonai.networks.layers.factoriesrrrrrr monai.utilsr __all__r r rrr3rrrr"s