488945b0a03843dc46953f801c6824d716d8fe9c,kornia/contrib/extract_patches.py,,extract_tensor_patches,#Any#Any#Any#Any#,127

Before Change

    See :class:`~kornia.contrib.ExtractTensorPatches` for details.
    
    return ExtractTensorPatches(window_size, stride, padding)(input)

After Change

    See :class:`~kornia.contrib.ExtractTensorPatches` for details.
    
    if not torch.is_tensor(input):
        raise TypeError("Input input type is not a torch.Tensor. Got {}"
                        .format(type(input)))
    if not len(input.shape) == 4:
        raise ValueError("Invalid input shape, we expect BxCxHxW. Got: {}"
                         .format(input.shape))

    if padding:
        pad_vert, pad_horz = _pair(padding)
        input = F.pad(input, [pad_horz, pad_horz, pad_vert, pad_vert])

    return _extract_tensor_patchesnd(input, _pair(window_size), _pair(stride))

Instances