morpheus.messages.multi_response_message.ResponseMemoryProbs
- class ResponseMemoryProbs(*args, **kwargs)[source]
Bases:
morpheus.messages.multi_response_message.ResponseMemory
Subclass of
ResponseMemory
containng an output tensor named ‘probs’.- Parameters
- probscupy.ndarray
Probabilities tensor
- Attributes
- probs
Methods
get_output
(name)Return the output tensor specified by
name
.- get_output(name)[source]
Return the output tensor specified by
name
.- Parameters
- namestr
Name of output tensor.
- Returns
- cupy.ndarray
Tensor corresponding to name.