def get_output_dtype(self, index):
        """
        Get the expected dtype of an output.
        """
        output = ctypes.c_int()
        self.DLL.AITemplateModelContainerGetOutputDtype(
            self.handle, index, ctypes.byref(output)
        )
        return output.value

        Get the expected dtype of an output.
        
python/aitemplate/compiler/model.py