nn module pytorch

Prosper Ndebele answered on April 12, 2024 Popularity 3/10 Helpfulness 1/10

answer nn module pytorch

nn module pytorch

Comment

import torch.nn as nn
import torch.nn.functional as F

class Model(nn.Module):
    def __init__(self):
        super().__init__()
        self.conv1 = nn.Conv2d(1, 20, 5)
        self.conv2 = nn.Conv2d(20, 20, 5)

    def forward(self, x):
        x = F.relu(self.conv1(x))
        return F.relu(self.conv2(x))

Popularity 3/10 Helpfulness 1/10 Language python

Source: pytorch.org

Tags: module python pytorch

Link to this answer
Share Copy Link

Contributed on Apr 12 2024

Prosper Ndebele

0 Answers Avg Quality 2/10

nn module pytorch

Contents

More Related Answers

nn module pytorch

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.