Skip to content

↔ [Converter] Add support for aten::Int.Tensor in Torch-TensorRT TorchScript Path #1836

@gs-olive

Description

@gs-olive

aten::Int.Tensor

  • Function Schema: aten::Int.Tensor(Tensor a) -> int

  • Original PyTorch API: torch.item() [documentation]

This converter/evaluator extracts the integer value from single-element tensors.

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions