# Codeblock 1
import torch
import torch.nn as nn
import torch.nn.functional as F

GRIDS = [4, 2, 1]