tugouxp的专栏

import torch.nn as nn import torch.nn.functional as F import torch device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d.