Belle II Software development
SNConv2d Class Reference
Inheritance diagram for SNConv2d:
SN

Public Member Functions

def __init__ (self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, num_svs=1, num_itrs=1, eps=1e-12)
 Constructor.
 
def forward (self, x)
 forward
 

Detailed Description

2D Conv layer with spectral norm

Definition at line 284 of file ieagan.py.

Constructor & Destructor Documentation

◆ __init__()

def __init__ (   self,
  in_channels,
  out_channels,
  kernel_size,
  stride = 1,
  padding = 0,
  dilation = 1,
  groups = 1,
  bias = True,
  num_svs = 1,
  num_itrs = 1,
  eps = 1e-12 
)

Constructor.

Reimplemented from SN.

Definition at line 290 of file ieagan.py.

303 ):
304 nn.Conv2d.__init__(
305 self,
306 in_channels,
307 out_channels,
308 kernel_size,
309 stride,
310 padding,
311 dilation,
312 groups,
313 bias,
314 )
315 SN.__init__(self, num_svs, num_itrs, out_channels, eps=eps)
316

Member Function Documentation

◆ forward()

def forward (   self,
  x 
)

forward

Definition at line 318 of file ieagan.py.

318 def forward(self, x):
319 return F.conv2d(
320 x,
321 self.W_(),
322 self.bias,
323 self.stride,
324 self.padding,
325 self.dilation,
326 self.groups,
327 )
328
329

The documentation for this class was generated from the following file: