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(TensorIndex count, TensorObject &&probs)
Construct a new Response Memory Probs object.
- Parameters
count –
probs –
-
ResponseMemoryProbs(TensorIndex count, TensorMap &&tensors)
Construct a new Response Memory Probs object.
- Parameters
count –
tensors –
-
const TensorObject &get_probs() const
Returns the tensor named ‘probs’. alias for
get_tensor("probs")
- Throws
- Returns
std::runtime_error – If no tensor named “probs” exists
const TensorObject&
-
void set_probs(TensorObject &&probs)
Update the tensor named ‘probs’.
- Parameters
- Throws
probs –
std::length_error – If the number of rows in
probs
does not matchcount
.
-
ResponseMemoryProbs(TensorIndex count, TensorObject &&probs)