Searched refs:nn (Results 1 – 1 of 1) sorted by relevance
3 import torch.nn as nn namespace20 class anisoSGS(nn.Module):28 self.net = nn.Sequential(29 nn.Linear(self.ndIn, self.nNeurons),30 nn.LeakyReLU(0.3),31 nn.Linear(self.nNeurons, self.ndOut))