from collections import deque

import torch
from torch import Tensor, nn