sglang_v0.5.2/pytorch_2.8.0/test/fx/named_tup.py

9 lines
116 B
Python

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