output_dtype

Function Usage

Obtains the data type of the output tensor.

Prototype

1
output_dtype(index: int)

Input Parameters

Parameter

Type

Description

index

int

Index of the model output (the indexth output)

Response Parameters

Data type (base.dtype) of the output tensor