experimental_experiment.xbuilder.model_container

class experimental_experiment.xbuilder.model_container.TorchModelContainer(*args, **kwargs)

Overwrites onnx.model_container.ModelContainer to support torch tensors.

experimental_experiment.xbuilder.model_container.proto_from_array(arr: torch.Tensor, name: str | None = None, verbose: int = 0) → TensorProto

Converts a torch Tensor into a TensorProto.