get_output_info
- Module.get_output_info(name)
Gets the information about the output tensor of model inference, such as tensor shape and data type with the given output tensor name.
- Parameters:
name (str) -- The name of the output tensor.
- Returns:
input_info -- The information of the output tensor.
- Return type:
Examples
See example in
Module.