from typing import NamedTuple

import torch

class MyNamedTup(NamedTuple):
    i : torch.Tensor
    f : torch.Tensor