# Glue layer for transforming whole pictures into 384x384 sequence for encoder # input import torch import torch.nn as nn import torch.nn.functional as F import torchvision from torchvision import transforms import numpy as np