strict_fusion

class torch.jit.strict_fusion [source]

Этот класс возвращает ошибку, если не все узлы были объединены во время вывода или символически дифференцированы во время обучения.

Пример:

Принудительное объединение сложений.

@torch.jit.script
def foo(x):
    with torch.jit.strict_fusion():
        return x + x + x