1from typing import NamedTuple 2 3import torch 4 5 6class MyNamedTup(NamedTuple): 7 i: torch.Tensor 8 f: torch.Tensor 9