# Codeblock 1
import torch
import torch.nn as nn