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 –
-
ResponseMemoryProbs(size_t count, TensorObject probs)