torch.fx.passes.tools_common.is_node_output_tensor

torch.fx.passes.tools_common.is_node_output_tensor(node) [исходный код]

Проверяет, возвращает ли узел Tensor.

ПРИМЕЧАНИЕ: перед вызовом этой функции необходимо выполнить ShapeProp для содержащего графа fx. Это связано с тем, что функция проверяет метаданные type узла. Эти метаданные создаются с помощью ShapeProp.

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

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

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

© 2026, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://docs.pytorch.org/docs/2.14/generated/torch.fx.passes.tools_common.is_node_output_tensor.html