Open shairubin opened 7 months ago
_TensorOrTensorsOrGradEdge = Union[ torch.Tensor, Sequence[torch.Tensor], "torch.autograd.graph.GradientEdge", Sequence["torch.autograd.graph.GradientEdge"]]
_TensorOrTensorsOrGradEdge = Union[ torch.Tensor, Sequence[torch.Tensor], "torch.autograd.graph.GradientEdge", Sequence["torch.autograd.graph.GradientEdge"]]