O que é nn linear em PyTorch?
O que é nn linear em PyTorch?

Vídeo: O que é nn linear em PyTorch?

Vídeo: O que é nn linear em PyTorch?
Vídeo: Torch.nn.Linear Module explained 2024, Marcha
Anonim

Da documentação: tocha CLASSE. nn . Linear (in_features, out_features, bias = True) Aplica um linear transformação para os dados de entrada: y = xW ^ T + b. Parâmetros: in_features - tamanho de cada amostra de entrada.

Da mesma forma, pergunta-se: como funciona o NN linear?

Linear . Aplica um linear transformação para os dados de entrada, ou seja, // y = Ax + b //. O tensor de entrada dado em forward (entrada) deve ser um vetor (tensor 1D) ou uma matriz (tensor 2D). Se a entrada for uma matriz, cada linha será considerada uma amostra de entrada de determinado lote.

Da mesma forma, o que é conv2d em PyTorch? conv2d (input, peso, self. preconceito, self. stride, self. padding, self. dilatação, self.

Em segundo lugar, o que é PyTorch nn?

PyTorch : nn o nn pacote define um conjunto de Módulos, que você pode pensar como uma camada de rede neural que produz saída a partir da entrada e pode ter alguns pesos treináveis. importar tocha # N é o tamanho do lote; D_in é a dimensão de entrada; # H é uma dimensão oculta; D_out é a dimensão de saída.

Como você usa o ReLU no PyTorch?

No PyTorch , você pode construir um ReLU camada usando a função simples relu1 = nn. ReLU com o argumento inplace = False. Desde o ReLU função é aplicada elemento a elemento, não há necessidade de especificar dimensões de entrada ou saída. O argumento inplace determina como a função trata a entrada.

Recomendado: