Lines Matching full:nn
4 torch.nn
6 .. automodule:: torch.nn
7 .. automodule:: torch.nn.modules
11 .. contents:: torch.nn
17 .. currentmodule:: torch.nn
47 .. currentmodule:: torch.nn.modules.module
71 nn.Conv1d
72 nn.Conv2d
73 nn.Conv3d
74 nn.ConvTranspose1d
75 nn.ConvTranspose2d
76 nn.ConvTranspose3d
77 nn.LazyConv1d
78 nn.LazyConv2d
79 nn.LazyConv3d
80 nn.LazyConvTranspose1d
81 nn.LazyConvTranspose2d
82 nn.LazyConvTranspose3d
83 nn.Unfold
84 nn.Fold
94 nn.MaxPool1d
95 nn.MaxPool2d
96 nn.MaxPool3d
97 nn.MaxUnpool1d
98 nn.MaxUnpool2d
99 nn.MaxUnpool3d
100 nn.AvgPool1d
101 nn.AvgPool2d
102 nn.AvgPool3d
103 nn.FractionalMaxPool2d
104 nn.FractionalMaxPool3d
105 nn.LPPool1d
106 nn.LPPool2d
107 nn.LPPool3d
108 nn.AdaptiveMaxPool1d
109 nn.AdaptiveMaxPool2d
110 nn.AdaptiveMaxPool3d
111 nn.AdaptiveAvgPool1d
112 nn.AdaptiveAvgPool2d
113 nn.AdaptiveAvgPool3d
123 nn.ReflectionPad1d
124 nn.ReflectionPad2d
125 nn.ReflectionPad3d
126 nn.ReplicationPad1d
127 nn.ReplicationPad2d
128 nn.ReplicationPad3d
129 nn.ZeroPad1d
130 nn.ZeroPad2d
131 nn.ZeroPad3d
132 nn.ConstantPad1d
133 nn.ConstantPad2d
134 nn.ConstantPad3d
135 nn.CircularPad1d
136 nn.CircularPad2d
137 nn.CircularPad3d
147 nn.ELU
148 nn.Hardshrink
149 nn.Hardsigmoid
150 nn.Hardtanh
151 nn.Hardswish
152 nn.LeakyReLU
153 nn.LogSigmoid
154 nn.MultiheadAttention
155 nn.PReLU
156 nn.ReLU
157 nn.ReLU6
158 nn.RReLU
159 nn.SELU
160 nn.CELU
161 nn.GELU
162 nn.Sigmoid
163 nn.SiLU
164 nn.Mish
165 nn.Softplus
166 nn.Softshrink
167 nn.Softsign
168 nn.Tanh
169 nn.Tanhshrink
170 nn.Threshold
171 nn.GLU
181 nn.Softmin
182 nn.Softmax
183 nn.Softmax2d
184 nn.LogSoftmax
185 nn.AdaptiveLogSoftmaxWithLoss
195 nn.BatchNorm1d
196 nn.BatchNorm2d
197 nn.BatchNorm3d
198 nn.LazyBatchNorm1d
199 nn.LazyBatchNorm2d
200 nn.LazyBatchNorm3d
201 nn.GroupNorm
202 nn.SyncBatchNorm
203 nn.InstanceNorm1d
204 nn.InstanceNorm2d
205 nn.InstanceNorm3d
206 nn.LazyInstanceNorm1d
207 nn.LazyInstanceNorm2d
208 nn.LazyInstanceNorm3d
209 nn.LayerNorm
210 nn.LocalResponseNorm
211 nn.RMSNorm
221 nn.RNNBase
222 nn.RNN
223 nn.LSTM
224 nn.GRU
225 nn.RNNCell
226 nn.LSTMCell
227 nn.GRUCell
237 nn.Transformer
238 nn.TransformerEncoder
239 nn.TransformerDecoder
240 nn.TransformerEncoderLayer
241 nn.TransformerDecoderLayer
251 nn.Identity
252 nn.Linear
253 nn.Bilinear
254 nn.LazyLinear
264 nn.Dropout
265 nn.Dropout1d
266 nn.Dropout2d
267 nn.Dropout3d
268 nn.AlphaDropout
269 nn.FeatureAlphaDropout
279 nn.Embedding
280 nn.EmbeddingBag
290 nn.CosineSimilarity
291 nn.PairwiseDistance
301 nn.L1Loss
302 nn.MSELoss
303 nn.CrossEntropyLoss
304 nn.CTCLoss
305 nn.NLLLoss
306 nn.PoissonNLLLoss
307 nn.GaussianNLLLoss
308 nn.KLDivLoss
309 nn.BCELoss
310 nn.BCEWithLogitsLoss
311 nn.MarginRankingLoss
312 nn.HingeEmbeddingLoss
313 nn.MultiLabelMarginLoss
314 nn.HuberLoss
315 nn.SmoothL1Loss
316 nn.SoftMarginLoss
317 nn.MultiLabelSoftMarginLoss
318 nn.CosineEmbeddingLoss
319 nn.MultiMarginLoss
320 nn.TripletMarginLoss
321 nn.TripletMarginWithDistanceLoss
331 nn.PixelShuffle
332 nn.PixelUnshuffle
333 nn.Upsample
334 nn.UpsamplingNearest2d
335 nn.UpsamplingBilinear2d
345 nn.ChannelShuffle
349 .. automodule:: torch.nn.parallel
357 nn.DataParallel
358 nn.parallel.DistributedDataParallel
362 .. automodule:: torch.nn.utils
364 From the ``torch.nn.utils`` module:
368 .. currentmodule:: torch.nn.utils
450 in :func:`torch.nn.utils.parameterize.register_parametrization`.
499 nn.utils.rnn.PackedSequence
500 nn.utils.rnn.pack_padded_sequence
501 nn.utils.rnn.pad_packed_sequence
502 nn.utils.rnn.pad_sequence
503 nn.utils.rnn.pack_sequence
504 nn.utils.rnn.unpack_sequence
505 nn.utils.rnn.unpad_sequence
512 nn.Flatten
513 nn.Unflatten
530 nn.modules.lazy.LazyModuleMixin
535 The following are aliases to their counterparts in ``torch.nn``:
543 nn.modules.normalization.RMSNorm
547 .. py:module:: torch.nn.backends
548 .. py:module:: torch.nn.utils.stateless
549 .. py:module:: torch.nn.backends.thnn
550 .. py:module:: torch.nn.common_types
551 .. py:module:: torch.nn.cpp
552 .. py:module:: torch.nn.functional
553 .. py:module:: torch.nn.grad
554 .. py:module:: torch.nn.init
555 .. py:module:: torch.nn.modules.activation
556 .. py:module:: torch.nn.modules.adaptive
557 .. py:module:: torch.nn.modules.batchnorm
558 .. py:module:: torch.nn.modules.channelshuffle
559 .. py:module:: torch.nn.modules.container
560 .. py:module:: torch.nn.modules.conv
561 .. py:module:: torch.nn.modules.distance
562 .. py:module:: torch.nn.modules.dropout
563 .. py:module:: torch.nn.modules.flatten
564 .. py:module:: torch.nn.modules.fold
565 .. py:module:: torch.nn.modules.instancenorm
566 .. py:module:: torch.nn.modules.lazy
567 .. py:module:: torch.nn.modules.linear
568 .. py:module:: torch.nn.modules.loss
569 .. py:module:: torch.nn.modules.module
570 .. py:module:: torch.nn.modules.normalization
571 .. py:module:: torch.nn.modules.padding
572 .. py:module:: torch.nn.modules.pixelshuffle
573 .. py:module:: torch.nn.modules.pooling
574 .. py:module:: torch.nn.modules.rnn
575 .. py:module:: torch.nn.modules.sparse
576 .. py:module:: torch.nn.modules.transformer
577 .. py:module:: torch.nn.modules.upsampling
578 .. py:module:: torch.nn.modules.utils
579 .. py:module:: torch.nn.parallel.comm
580 .. py:module:: torch.nn.parallel.distributed
581 .. py:module:: torch.nn.parallel.parallel_apply
582 .. py:module:: torch.nn.parallel.replicate
583 .. py:module:: torch.nn.parallel.scatter_gather
584 .. py:module:: torch.nn.parameter
585 .. py:module:: torch.nn.utils.clip_grad
586 .. py:module:: torch.nn.utils.convert_parameters
587 .. py:module:: torch.nn.utils.fusion
588 .. py:module:: torch.nn.utils.init
589 .. py:module:: torch.nn.utils.memory_format
590 .. py:module:: torch.nn.utils.parametrizations
591 .. py:module:: torch.nn.utils.parametrize
592 .. py:module:: torch.nn.utils.prune
593 .. py:module:: torch.nn.utils.rnn