File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -900,13 +900,13 @@ class GaussianSmoothing(nn.Module):
900900 in the input using a depthwise convolution.
901901
902902 Args:
903- channels (`int` or `sequence`):
903+ channels (`int` or `sequence`):
904904 Number of channels of the input tensors. The output will have this number of channels as well.
905- kernel_size (`int` or `sequence`):
905+ kernel_size (`int` or `sequence`):
906906 Size of the Gaussian kernel.
907- sigma (`float` or `sequence`):
907+ sigma (`float` or `sequence`):
908908 Standard deviation of the Gaussian kernel.
909- dim (`int`, *optional*, defaults to `2`):
909+ dim (`int`, *optional*, defaults to `2`):
910910 The number of dimensions of the data. Default is 2 (spatial dimensions).
911911 """
912912
@@ -953,7 +953,7 @@ def forward(self, input):
953953 Input to apply Gaussian filter on.
954954
955955 Returns:
956- `torch.Tensor`:
956+ `torch.Tensor`:
957957 The filtered output tensor with the same shape as the input.
958958 """
959959 return self .conv (input , weight = self .weight .to (input .dtype ), groups = self .groups , padding = "same" )
You can’t perform that action at this time.
0 commit comments