Что такое nn linear в PyTorch?

2024 Автор: Lynn Donovan | [email protected]. Последнее изменение: 2023-12-15 23:52

Из документации: фонарик КЛАСС. nn . Линейный (in_features, out_features, bias = True) Применяет линейный преобразование во входящие данные: y = xW ^ T + b. Параметры: in_features - размер каждой входной выборки.

Точно так же спрашивается, как работает NN linear?

Линейный . Применяет линейный преобразование во входящие данные, т.е. // y = Ax + b //. Входной тензор, заданный в forward (вход), должен быть либо вектором (1D тензор), либо матрицей (2D тензор). Если входными данными является матрица, то предполагается, что каждая строка является входной выборкой данной партии.

Аналогично, что такое conv2d в PyTorch? conv2d (ввод, вес, себя. пристрастие, само. шаг, сам.

Во-вторых, что такое PyTorch nn?

PyTorch : nn В nn package определяет набор модулей, которые вы можете рассматривать как слой нейронной сети, который производит вывод из ввода и может иметь некоторые обучаемые веса. import torch # N - размер партии; D_in - входной размер; # H - скрытое измерение; D_out - размер вывода.

Как вы используете ReLU в PyTorch?

В PyTorch , вы можете построить ReLU слой с использованием простая функция relu1 = nn. ReLU с аргументом inplace = False. Поскольку ReLU функция применяется поэлементно, нет необходимости указывать входные или выходные размеры. Аргумент inplace определяет, как функция обрабатывает ввод.