Python torch.nn Module

This page shows the popular functions and classes defined in the torch.nn module. The items are ordered by their popularity in 40,000 open source Python projects. If you can not find a good example below, you can try the search function to search modules.

• 1. Module()    
  Used in 3029 projects
• 2. Linear()    
  Used in 2500 projects
• 3. Sequential()    
  Used in 2146 projects
• 4. Conv2d()    
  Used in 2038 projects
• 5. ReLU()    
  Used in 1959 projects
• 6. BatchNorm2d()    
  Used in 1583 projects
• 7. Dropout()    
  Used in 1174 projects
• 8. ModuleList()    
  Used in 1125 projects
• 9. MaxPool2d()    
  Used in 1123 projects
• 10. Parameter()    
  Used in 1097 projects
• 11. CrossEntropyLoss()    
  Used in 973 projects
• 12. Embedding()    
  Used in 719 projects
• 13. AvgPool2d()    
  Used in 690 projects
• 14. Sigmoid()    
  Used in 661 projects
• 15. ConvTranspose2d()    
  Used in 619 projects
• 16. BatchNorm1d()    
  Used in 586 projects
• 17. DataParallel()    
  Used in 565 projects
• 18. LeakyReLU()    
  Used in 550 projects
• 19. Tanh()    
  Used in 537 projects
• 20. MSELoss()    
  Used in 515 projects
• 21. LSTM    
  Used in 507 projects
• 22. Softmax()    
  Used in 499 projects
• 23. AdaptiveAvgPool2d()    
  Used in 405 projects
• 24. Conv1d()    
  Used in 372 projects
• 25. GRU    
  Used in 337 projects
• 26. Upsample()    
  Used in 337 projects
• 27. Dropout2d()    
  Used in 277 projects
• 28. BCELoss()    
  Used in 227 projects
• 29. LogSoftmax()    
  Used in 220 projects
• 30. BCEWithLogitsLoss()    
  Used in 207 projects
• 31. NLLLoss()    
  Used in 207 projects
• 32. Conv3d()    
  Used in 200 projects
• 33. LayerNorm()    
  Used in 179 projects
• 34. LSTMCell()    
  Used in 177 projects
• 35. BatchNorm3d()    
  Used in 174 projects
• 36. PReLU()    
  Used in 174 projects
• 37. GroupNorm()    
  Used in 171 projects
• 38. L1Loss()    
  Used in 153 projects
• 39. InstanceNorm2d()    
  Used in 151 projects
• 40. ELU    
  Used in 132 projects
• 41. ReLU6()    
  Used in 125 projects
• 42. ParameterList()    
  Used in 107 projects
• 43. MaxPool1d()    
  Used in 107 projects
• 44. ReflectionPad2d()    
  Used in 105 projects
• 45. MaxPool3d()    
  Used in 103 projects
• 46. GRUCell()    
  Used in 100 projects
• 47. RNN    
  Used in 95 projects
• 48. ZeroPad2d()    
  Used in 83 projects
• 49. UpsamplingBilinear2d()    
  Used in 78 projects
• 50. AdaptiveMaxPool2d()    
  Used in 77 projects
• 51. KLDivLoss()    
  Used in 74 projects
• 52. ConvTranspose3d()    
  Used in 67 projects
• 53. PixelShuffle()    
  Used in 67 projects
• 54. Softplus()    
  Used in 64 projects
• 55. SmoothL1Loss()    
  Used in 63 projects
• 56. Identity()    
  Used in 61 projects
• 57. ModuleDict()    
  Used in 59 projects
• 58. ReplicationPad2d()    
  Used in 58 projects
• 59. AvgPool3d()    
  Used in 56 projects
• 60. html()    
  Used in 53 projects
• 61. functional()    
  Used in 49 projects
• 62. Hardtanh()    
  Used in 46 projects
• 63. SELU    
  Used in 42 projects
• 64. ConvTranspose1d()    
  Used in 40 projects
• 65. AvgPool1d()    
  Used in 38 projects
• 66. MarginRankingLoss()    
  Used in 37 projects
• 67. InstanceNorm1d()    
  Used in 37 projects
• 68. MaxUnpool2d()    
  Used in 34 projects
• 69. init()    
  Used in 32 projects
• 70. SyncBatchNorm()    
  Used in 31 projects
• 71. AdaptiveAvgPool3d()    
  Used in 30 projects
• 72. NLLLoss2d()    
  Used in 30 projects
• 73. AdaptiveAvgPool1d()    
  Used in 28 projects
• 74. AdaptiveMaxPool1d()    
  Used in 27 projects
• 75. UpsamplingNearest2d()    
  Used in 27 projects
• 76. Bilinear()    
  Used in 26 projects
• 77. Dropout3d()    
  Used in 26 projects
• 78. ConstantPad2d()    
  Used in 25 projects
• 79. parallel()    
  Used in 25 projects
• 80. Softmax2d()    
  Used in 22 projects
• 81. InstanceNorm3d()    
  Used in 22 projects
• 82. GLU    
  Used in 21 projects
• 83. RNNCell()    
  Used in 19 projects
• 84. CosineSimilarity()    
  Used in 18 projects
• 85. Flatten()    
  Used in 17 projects
• 86. SoftMarginLoss()    
  Used in 17 projects
• 87. LogSigmoid()    
  Used in 16 projects
• 88. MultiLabelSoftMarginLoss()    
  Used in 16 projects
• 89. CosineEmbeddingLoss()    
  Used in 15 projects
• 90. RReLU()    
  Used in 15 projects
• 91. ConstantPad1d()    
  Used in 14 projects
• 92. AlphaDropout()    
  Used in 13 projects
• 93. TripletMarginLoss()    
  Used in 13 projects
• 94. CTCLoss()    
  Used in 13 projects
• 95. LocalResponseNorm()    
  Used in 13 projects
• 96. Threshold()    
  Used in 12 projects
• 97. Softsign()    
  Used in 11 projects
• 98. module()    
  Used in 10 projects
• 99. TransformerEncoder()    
  Used in 10 projects
• 100. TransformerEncoderLayer()    
  Used in 10 projects
• 101. Sequentail()    
  Used in 9 projects
• 102. AdaptiveLogSoftmaxWithLoss()    
  Used in 9 projects
• 103. AdaptiveMaxPool3d()    
  Used in 9 projects
• 104. MultiheadAttention()    
  Used in 9 projects
• 105. EmbeddingBag()    
  Used in 9 projects
• 106. ParameterDict()    
  Used in 8 projects
• 107. HingeEmbeddingLoss()    
  Used in 8 projects
• 108. MultiLabelMarginLoss()    
  Used in 8 projects
• 109. MultiMarginLoss()    
  Used in 8 projects
• 110. Conv2DBatchNorm()    
  Used in 7 projects
• 111. Softshrink()    
  Used in 7 projects
• 112. CELU    
  Used in 7 projects
• 113. Tanhshrink()    
  Used in 7 projects
• 114. DistributedDataParallel()    
  Used in 7 projects
• 115. MaxUnpool3d()    
  Used in 7 projects
• 116. Modules()    
  Used in 6 projects
• 117. PoissonNLLLoss()    
  Used in 6 projects
• 118. ConstantPad3d()    
  Used in 6 projects
• 119. ReflectionPad1d()    
  Used in 6 projects
• 120. deconv2DBatchNormRelu()    
  Used in 6 projects
• 121. Hardshrink()    
  Used in 6 projects
• 122. Softmin()    
  Used in 5 projects
• 123. ReplicationPad3d()    
  Used in 5 projects
• 124. __dict__()    
  Used in 5 projects
• 125. Unfold()    
  Used in 5 projects
• 126. RNNBase()    
  Used in 5 projects
• 127. ReplicationPad1d()    
  Used in 5 projects