output_dtype(index)
函数功能
获得模型输出的对应Tensor的数据类型信息。
输入参数说明
参数名 |
类型 |
说明 |
---|---|---|
index |
int |
模型输出的索引(第index个输出)。 |
返回参数说明
返回输出Tensor的数据类型,base.dtype类型。
父主题: Model类
获得模型输出的对应Tensor的数据类型信息。
参数名 |
类型 |
说明 |
---|---|---|
index |
int |
模型输出的索引(第index个输出)。 |
返回输出Tensor的数据类型,base.dtype类型。