view_as_windows_torch

view_as_windows_torch#

jolideco.utils.torch.view_as_windows_torch(image, shape, stride)[source]#

View tensor as overlapping rectangular windows

Parameters:
  • image (~torch.Tensor) – Image tensor

  • shape (tuple) – Shape of the patches.

  • stride (int) – Stride of the patches. By default it is half of the patch size.

Returns:

windows – Tensor of overlapping windows

Return type:

~torch.Tensor