Псевдонимы в torch.nn
Создано: 25 июля 2025 г. | Последнее обновление: 8 мая 2026 г.
Ниже приведены псевдонимы соответствующих объектов в torch.nn во вложенных пространствах имён.
torch.nn.modules
Ниже приведены псевдонимы соответствующих объектов в torch.nn в пространстве имён torch.nn.modules.
Контейнеры (псевдонимы)
Последовательный контейнер. | |
Содержит подмодули в списке. | |
Содержит подмодули в словаре. | |
Содержит параметры в списке. | |
Содержит параметры в словаре. |
Сверточные слои (псевдонимы)
Применяет одномерную свертку к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет двумерную свертку к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет трехмерную свертку к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет оператор одномерной транспонированной свертки к входному изображению, состоящему из нескольких входных плоскостей. | |
Применяет оператор двумерной транспонированной свертки к входному изображению, состоящему из нескольких входных плоскостей. | |
Применяет оператор трехмерной транспонированной свертки к входному изображению, состоящему из нескольких входных плоскостей. | |
Модуль | |
Модуль | |
Модуль | |
Модуль | |
Модуль | |
Модуль | |
Извлекает локальные блоки со скользящим окном из входного тензора с пакетами. | |
Объединяет массив локальных блоков со скользящим окном в большой содержащий их тензор. |
Слои пулинга (псевдонимы)
Применяет одномерный максимальный пулинг к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет двумерный максимальный пулинг к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет трехмерный максимальный пулинг к входному сигналу, состоящему из нескольких входных плоскостей. | |
Вычисляет частичное обратное преобразование | |
Вычисляет частичное обратное преобразование | |
Вычисляет частичное обратное преобразование | |
Применяет одномерный пулинг со средним значением к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет двумерный пулинг со средним значением к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет трехмерный пулинг со средним значением к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет двумерный дробный максимальный пулинг к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет трехмерный дробный максимальный пулинг к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет одномерный степенной средний пулинг к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет двумерный степенной средний пулинг к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет трехмерный степенной средний пулинг к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет адаптивный одномерный максимальный пулинг к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет адаптивный двумерный максимальный пулинг к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет адаптивный трехмерный максимальный пулинг к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет адаптивный одномерный пулинг со средним значением к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет адаптивный двумерный пулинг со средним значением к входному сигналу, состоящему из нескольких входных плоскостей. | |
Применяет адаптивный трехмерный пулинг со средним значением к входному сигналу, состоящему из нескольких входных плоскостей. |
Слои дополнения (псевдонимы)
Дополняет входной тензор, отражая границы входных данных. | |
Дополняет входной тензор, отражая границы входных данных. | |
Дополняет входной тензор, отражая границы входных данных. | |
Дополняет входной тензор, повторяя значения на границах входных данных. | |
Дополняет входной тензор, повторяя значения на границах входных данных. | |
Дополняет входной тензор, повторяя значения на границах входных данных. | |
Дополняет границы входного тензора нулями. | |
Дополняет границы входного тензора нулями. | |
Дополняет границы входного тензора нулями. | |
Дополняет границы входного тензора постоянным значением. | |
Дополняет границы входного тензора постоянным значением. | |
Дополняет границы входного тензора постоянным значением. | |
Дополняет входной тензор циклическим способом, используя значения с противоположных границ входных данных. | |
Дополняет входной тензор циклическим способом, используя значения с противоположных границ входных данных. | |
Дополняет входной тензор циклическим способом, используя значения с противоположных границ входных данных. |
Нелинейные функции активации (взвешенная сумма, нелинейность) (псевдонимы)
Поэлементно применяет функцию экспоненциальной линейной единицы (ELU). | |
Поэлементно применяет функцию жесткого сжатия (Hardshrink). | |
Поэлементно применяет функцию Hardsigmoid. | |
Поэлементно применяет функцию HardTanh. | |
Поэлементно применяет функцию Hardswish. | |
Поэлементно применяет функцию LeakyReLU. | |
Поэлементно применяет функцию Logsigmoid. | |
Позволяет модели совместно учитывать информацию из различных подпространств представлений. | |
Поэлементно применяет функцию PReLU. | |
Поэлементно применяет функцию выпрямленной линейной единицы. | |
Поэлементно применяет функцию ReLU6. | |
Поэлементно применяет функцию рандомизированной Leaky ReLU. | |
Поэлементно применяет функцию SELU. | |
Поэлементно применяет функцию CELU. | |
Применяет функцию гауссовых линейных единиц. | |
Поэлементно применяет функцию Sigmoid. | |
Поэлементно применяет функцию сигмоидальной линейной единицы (SiLU). | |
Поэлементно применяет функцию Mish. | |
Поэлементно применяет функцию Softplus. | |
Поэлементно применяет функцию мягкого сжатия. | |
Поэлементно применяет функцию Softsign. | |
Поэлементно применяет функцию гиперболического тангенса (Tanh). | |
Поэлементно применяет функцию Tanhshrink. | |
Применяет пороговое преобразование к каждому элементу входного тензора. | |
Применяет функцию управляемой линейной единицы. |
Нелинейные функции активации (другие) (псевдонимы)
Применяет функцию Softmin к n-мерному входному тензору. | |
Применяет функцию Softmax к n-мерному входному тензору. | |
Применяет SoftMax к признакам в каждой пространственной точке. | |
Применяет функцию к n-мерному входному тензору. | |
Эффективное приближение Softmax. |
Слои нормализации (псевдонимы)
Выполняет пакетную нормализацию входных данных размерности 2 или 3. | |
Выполняет пакетную нормализацию входных данных размерности 4. | |
Выполняет пакетную нормализацию входных данных размерности 5. | |
Модуль | |
Модуль | |
Модуль | |
Выполняет групповую нормализацию мини-пакета входных данных. | |
Выполняет пакетную нормализацию входных данных размерности N. | |
Выполняет нормализацию экземпляров. | |
Выполняет нормализацию экземпляров. | |
Выполняет нормализацию экземпляров. | |
Модуль | |
Модуль | |
Модуль | |
Выполняет послойную нормализацию мини-пакета входных данных. | |
Выполняет локальную нормализацию отклика входного сигнала. | |
Выполняет послойную нормализацию по среднеквадратичному значению мини-пакета входных данных. |
Рекуррентные слои (псевдонимы)
Базовый класс модулей RNN (RNN, LSTM, GRU). | |
Применяет к входной последовательности многослойную рекуррентную нейронную сеть Элмана с нелинейностью или . | |
Применяет к входной последовательности многослойную рекуррентную нейронную сеть с долговременной краткосрочной памятью (LSTM). | |
Применяет к входной последовательности многослойную рекуррентную нейронную сеть с управляемыми рекуррентными блоками (GRU). | |
Ячейка рекуррентной нейронной сети Элмана с нелинейностью tanh или ReLU. | |
Ячейка с долговременной краткосрочной памятью (LSTM). | |
Ячейка с управляемыми рекуррентными блоками (GRU). |
Слои трансформера (псевдонимы)
Базовый слой трансформера. | |
TransformerEncoder представляет собой стек из N слоёв кодировщика. | |
TransformerDecoder представляет собой стек из N слоёв декодировщика. | |
TransformerEncoderLayer состоит из слоя самовнимания и нейронной сети прямого распространения. | |
TransformerDecoderLayer состоит из слоя самовнимания, слоя многоголового внимания и нейронной сети прямого распространения. |
Линейные слои (псевдонимы)
Заглушка для тождественного оператора, не зависящего от аргументов. | |
Применяет к входным данным аффинное линейное преобразование: . | |
Применяет к входным данным билинейное преобразование: . | |
Модуль |
Слои прореживания (псевдонимы)
Во время обучения случайным образом обнуляет некоторые элементы входного тензора с вероятностью | |
Случайным образом обнуляет целые каналы. | |
Случайным образом обнуляет целые каналы. | |
Случайным образом обнуляет целые каналы. | |
Применяет альфа-прореживание к входным данным. | |
Случайным образом маскирует целые каналы. |
Разреженные слои (псевдонимы)
Простая таблица поиска, в которой хранятся эмбеддинги фиксированного словаря и размера. | |
Вычисляет суммы или средние значения для «мешков» эмбеддингов, не создавая промежуточные эмбеддинги. |
Функции расстояния (псевдонимы)
Возвращает косинусное сходство между и , вычисленное вдоль | |
Вычисляет попарное расстояние между входными векторами или между столбцами входных матриц. |
Функции потерь (псевдонимы)
Создаёт функцию потерь, вычисляющую среднюю абсолютную ошибку (MAE) между каждым элементом входных данных и целевыми данными . | |
Создаёт функцию потерь, вычисляющую среднеквадратичную ошибку (квадрат нормы L2) между каждым элементом входных данных и целевыми данными . | |
Эта функция потерь вычисляет кросс-энтропию между входными логитами и целевыми данными. | |
Функция потерь Connectionist Temporal Classification (CTC). | |
Функция потерь по отрицательному логарифму правдоподобия. | |
Функция потерь по отрицательному логарифму правдоподобия с распределением Пуассона для целевых данных. | |
Функция потерь по отрицательному логарифму правдоподобия с распределением Гаусса. | |
Функция потерь, основанная на дивергенции Кульбака — Лейблера. | |
Создаёт функцию потерь, вычисляющую бинарную кросс-энтропию между целевыми данными и входными вероятностями: | |
Эта функция потерь объединяет слой | |
Создаёт функцию потерь для входных данных , — мини-пакета из двух одномерных или нульмерных | |
Вычисляет потери по входному тензору и тензору меток (содержащему 1 или -1). | |
Создаёт функцию потерь, оптимизирующую многоклассовую шарнирную функцию потерь (основанную на отступе) между входными данными (двумерным мини-пакетом | |
Создаёт функцию потерь, использующую квадратичный член, если абсолютная поэлементная ошибка меньше delta, и член L1, масштабированный на delta, в противном случае. | |
Создаёт функцию потерь, использующую квадратичный член, если абсолютная поэлементная ошибка меньше beta, и член L1 в противном случае. | |
Создаёт функцию потерь, оптимизирующую логистические потери для классификации на два класса между входным тензором и целевым тензором (содержащим 1 или -1). | |
Создаёт функцию потерь, оптимизирующую потери «один против всех» для нескольких меток на основе максимальной энтропии между входными данными и целевыми данными размера . | |
Создаёт функцию потерь для входных тензоров , и метки | |
Создаёт функцию потерь, оптимизирующую многоклассовую шарнирную функцию потерь (основанную на отступе) между входными данными (двумерным мини-пакетом | |
Создаёт функцию потерь для триплетов по входным тензорам , , и отступу со значением больше . | |
Создаёт функцию потерь для триплетов по входным тензорам , и (соответственно представляющих опорный, положительный и отрицательный примеры), а также неотрицательной вещественной функции («функции расстояния»), используемой для вычисления взаимосвязи между опорным и положительным примерами («положительное расстояние») и между опорным и отрицательным примерами («отрицательное расстояние»). |
Слои обработки изображений (псевдонимы)
Переставляет элементы тензора в соответствии с коэффициентом масштабирования. | |
Выполняет обратную операцию PixelShuffle. | |
Увеличивает разрешение заданных многоканальных одномерных (временных), двумерных (пространственных) или трёхмерных (объёмных) данных. | |
Применяет двумерное увеличение разрешения методом ближайшего соседа к входному сигналу, состоящему из нескольких каналов. | |
Применяет двумерное билинейное увеличение разрешения к входному сигналу, состоящему из нескольких каналов. |
Слои перестановки (псевдонимы)
Разделяет и переставляет каналы в тензоре. |
torch.nn.modules.utils
-
torch.nn.modules.utils.consume_prefix_in_state_dict_if_present(state_dict, prefix)[исходный код] -
Удаляет префикс из state_dict на месте, если он есть.
Примечание
Локальная модель может загрузить
state_dictмодели DP/DDP, предварительно применивconsume_prefix_in_state_dict_if_present(state_dict, “module.”)перед вызовомtorch.nn.Module.load_state_dict().- Параметры:
-
- state_dict (OrderedDict) – state-dict, который нужно загрузить в модель.
- prefix (str) – префикс.
torch.nn.utils
Ниже перечислены псевдонимы соответствующих объектов из torch.nn.utils во вложенных пространствах имён.
Вспомогательные функции для ограничения градиентов параметров.
Ограничивает норму градиента итерируемой последовательности параметров. | |
Ограничивает норму градиента итерируемой последовательности параметров. | |
Ограничивает градиенты итерируемой последовательности параметров заданным значением. |
Вспомогательные функции для преобразования параметров Module в один вектор и обратно.
Преобразует итерируемую последовательность параметров в один вектор. | |
Копирует срезы вектора в итерируемую последовательность параметров. |
Вспомогательные функции для объединения модулей с модулями BatchNorm.
Объединяет свёрточный модуль и модуль BatchNorm в один новый свёрточный модуль. | |
Объединяет параметры свёрточного модуля и модуля BatchNorm в параметры нового свёрточного модуля. | |
Объединяет линейный модуль и модуль BatchNorm в один новый линейный модуль. | |
Объединяет параметры линейного модуля и модуля BatchNorm в параметры нового линейного модуля. |
Вспомогательные функции для преобразования форматов памяти параметров Module.
Преобразует | |
Преобразует |
Вспомогательные функции для применения и удаления нормализации весов параметров Module.
Применяет нормализацию весов к параметру указанного модуля. | |
Удаляет параметризацию нормализации весов из модуля. | |
Применяет спектральную нормализацию к параметру указанного модуля. | |
Удаляет параметризацию спектральной нормализации из модуля. |
Вспомогательные функции для инициализации параметров Module.
Создаёт экземпляр класса модуля с заданными аргументами и именованными аргументами, не инициализируя параметры и буферы. |
© 2026, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://docs.pytorch.org/docs/2.14/nn.aliases.html