torch.fx.annotate.annotate

torch.fx.annotate.annotate(val, type) [исходный код]

Аннотирует объект Proxy заданным типом.

Эта функция аннотирует val заданным типом, если тип val — объект torch.fx.Proxy :param val: Объект, который нужно аннотировать, если его тип — torch.fx.Proxy. :type val: object :param type: Тип, который будет присвоен заданному объекту proxy в качестве val. :type type: object

Возвращает:

Заданный val.

Вызывает исключение:

RuntimeError – Если для val уже задан тип в его узле.

Тип возвращаемого значения:

Any

Предупреждение

Этот API является экспериментальным и НЕ обладает обратной совместимостью.