torch.jit.annotate

torch.jit.annotate(the_type, the_value) [source]

Этот метод является проходящей функцией, которая возвращает the_value, используемой для подсказки компилятору TorchScript типа the_value. Он является бесполезной операцией (no-op) при выполнении вне TorchScript.

Хотя TorchScript может определить правильный тип для большинства Python-выражений, существуют некоторые случаи, когда вывод типа может быть неправильным, включая:

• Пустые контейнеры, такие как [] и {}, которые TorchScript предполагает как контейнеры типа Tensor
• Типы по умолчанию, такие как Optional[T] , но присвоенное действительное значение типа T, TorchScript предположит, что это тип T вместо Optional[T]

Обратите внимание, что annotate() не помогает в __init__ методе подклассов torch.nn.Module , так как он выполняется в режиме жадного вычисления. Для аннотации типов torch.nn.Module атрибутов используйте Annotate() вместо этого.

Пример:

import torch
from typing import Dict

@torch.jit.script
def fn():
    # Telling TorchScript that this empty dictionary is a (str -> int) dictionary
    # instead of default dictionary type of (str -> Tensor).
    d = torch.jit.annotate(Dict[str, int], {})

    # Without `torch.jit.annotate` above, following statement would fail because of
    # type mismatch.
    d["name"] = 20

Параметры

• the_type – Тип Python, который должен быть передан компилятору TorchScript в качестве подсказки типа для the_value
• the_value – Значение или выражение для подсказки типа.

Возвращает

the_value возвращается как значение.