# Codeblock 1
import torch
import torch.nn as nn

S = 7
B = 2
C = 20