Method GetOutputDim

dali/pipeline/workspace/workspace.h:305–307

* Returns number of dimensions for a given output */

   * Returns number of dimensions for a given output
   */
  int GetOutputDim(int output_idx) const {
    return GetBufferProperty(outputs_, output_idx, [](auto &buf) { return buf->sample_dim(); });
  }

  /**
   * Returns batch size that the Operator is expected to produce on a given output