Method GetOutputBatchSize

dali/pipeline/workspace/workspace.h:291–293

* Returns batch size for a given output */

   * Returns batch size for a given output
   */
  int GetOutputBatchSize(int output_idx) const {
    return GetBufferProperty(outputs_, output_idx, [](auto &buf) { return buf->num_samples(); });
  }

  /**
   * Returns number of dimensions for a given input