Class ResponseMemoryProbs

• Defined in File response_memory_probs.hpp

Base Type

• public morpheus::ResponseMemory (Class ResponseMemory)

class ResponseMemoryProbs : public morpheus::ResponseMemory

Output memory block containing the inference response probabilities.

Public Functions

ResponseMemoryProbs(size_t count, TensorObject probs)

Construct a new Response Memory Probs object.

Parameters

• count –

• probs –

ResponseMemoryProbs(size_t count, tensor_map_t &&tensors)

Construct a new Response Memory Probs object.

Parameters

• count –

• tensors –

const TensorObject &get_probs() const

Returns the tensor named ‘probs’, throws a std::runtime_error if it does not exist.

Returns

const TensorObject&

void set_probs(TensorObject probs)

Update the tensor named ‘probs’.

Parameters

probs –