output_dtype(index)
Function Usage
Obtains the data type of the output tensor.
Input Parameter Description
Parameter |
Type |
Description |
|---|---|---|
index |
int |
Index of the model output (the indexth output) |
Return Parameter Description
Returns the data type (base.dtype) of the output tensor.
Parent topic: Model