This way:
import torch
from torch.nn import functional as f
x = torch.arange(1, 17).view(4, 4).float()
x = x[None, None, :, :]
f.unfold(x, kernel_size=2)

It prints:
tensor([[[ 1.,  2.,  3.,  5.,  6.,  7.,  9., 10., 11.],
         [ 2.,  3.,  4.,  6.,  7.,  8., 10., 11., 12.],
         [ 5.,  6.,  7.,  9., 10., 11., 13., 14., 15.],
         [ 6.,  7.,  8., 10., 11., 12., 14., 15., 16.]]])