from typing import NamedTuple
import torch
class MyNamedTup(NamedTuple):
i: torch.Tensor
f: torch.Tensor