Skip to content

Commit

Permalink
Update functional.py
Browse files Browse the repository at this point in the history
gaussian blur doc - mention padding policy
  • Loading branch information
talcs authored Feb 1, 2024
1 parent 806dba6 commit d6dbf6f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchvision/transforms/functional.py
Original file line number Diff line number Diff line change
Expand Up @@ -1301,7 +1301,7 @@ def erase(img: Tensor, i: int, j: int, h: int, w: int, v: Tensor, inplace: bool


def gaussian_blur(img: Tensor, kernel_size: List[int], sigma: Optional[List[float]] = None) -> Tensor:
"""Performs Gaussian blurring on the image by given kernel.
"""Performs Gaussian blurring on the image by given kernel, using reflection padding corresponding to the kernel size, to maintain the input shape.
If the image is torch Tensor, it is expected
to have [..., H, W] shape, where ... means at most one leading dimension.
Expand Down

0 comments on commit d6dbf6f

Please sign in to comment.