NA input tensor of shape \((\mbox{minibatch} , \mbox{in\_channels} , iW)\)
weight
NA filters of shape \((\mbox{in\_channels} , \frac{\mbox{out\_channels}}{\mbox{groups}} , kW)\)
bias
NA optional bias of shape \((\mbox{out\_channels})\). Default: None
stride
NA the stride of the convolving kernel. Can be a single number or a tuple (sW,). Default: 1
padding
NA dilation * (kernel_size - 1) - padding zero-padding will be added to both sides of each dimension in the input. Can be a single number or a tuple (padW,). Default: 0
output_padding
NA additional size added to one side of each dimension in the output shape. Can be a single number or a tuple (out_padW). Default: 0
groups
NA split input into groups, \(\mbox{in\_channels}\) should be divisible by the number of groups. Default: 1
dilation
NA the spacing between kernel elements. Can be a single number or a tuple (dW,). Default: 1