"""TBC."""
from typing import Dict

import torch
from torch import Tensor