Есть ли способ указать типовый DataLoader DataLoader Pytorch? ⇐ Python

[Anonymous]

Скажем, у меня есть следующее: < /p>

Код: Выделить всё

class SomeClass:
def some_function(dataloader: DataLoader):
for idx, batch in enumerate(dataloader):
... do something with batch ...

Я хотел бы ввести DataLoader так, чтобы я мог показать через функцию typemitnting, какой формат я ожидаю, что пакет будет иметь. Например, я хотел бы, чтобы Packation Be Type Tuple [Tensor, Tensor] или я хотел бы ввести его Tuple [Tensor, Tensor, CustomObject] . Есть ли способ, которым я могу указать это?>

Подробнее здесь: https://stackoverflow.com/questions/758 ... dataloader