TensorPlay
API symbolslibrary
Copy
View Markdown

tensorplay.library.infer_schema

tensorplay.library.infer_schema(prototype_function: Callable[[...], Any], /, *, mutates_args: Sequence[str], op_name: str | None = None, tags: Any = ()) str[source]

Produces "ns::op(Tensor self, SymInt n, Tensor(a!) out) -> Tensor" alias-annotation (<type>(<letter>!)) marker. Parameters without annotations are treated as tensors; *args/**kwargs are skipped.

On this page

Ask DeepWiki