|
|
|
@@ -32,7 +32,7 @@ class Conv2dBlock(nn.Cell): |
|
|
|
kernel_size (int): Input kernel size. Default: 1 |
|
|
|
stride (int): Stride size for the first convolutional layer. Default: 1. |
|
|
|
padding (int): Implicit paddings on both sides of the input. Default: 0. |
|
|
|
pad_mode (int): Padding mode. Optional values are "same", "valid", "pad". Default: "same". |
|
|
|
pad_mode (str): Padding mode. Optional values are "same", "valid", "pad". Default: "same". |
|
|
|
Returns: |
|
|
|
Tensor, output tensor. |
|
|
|
""" |
|
|
|
|