Class ResponseMemoryProbs

Base Type

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

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