Class ResponseMemoryProbs

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

std::runtime_error – If no tensor named “probs” exists

Returns

const TensorObject&

void set_probs(TensorObject &&probs)

Update the tensor named ‘probs’.

Parameters

probs

Throws

std::length_error – If the number of rows in probs does not match count.

Previous Class ResponseMemory
Next Class RESTDataLoader
© Copyright 2024, NVIDIA. Last updated on Apr 11, 2024.