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.

© Copyright 2023, NVIDIA. Last updated on Feb 3, 2023.