Method ValidateOutputShape

dali/kernels/common/block_setup.h:167–187

* @brief Check if `out_shape` matches the number of channels of `in_shape` and spatial dimensions * of `output_sizes` as if it was generated with GetOutputShape(in_shape, output_sizes). */

   * of `output_sizes` as if it was generated with GetOutputShape(in_shape, output_sizes).
   */
  void ValidateOutputShape(
      const TensorListShape<tensor_ndim> &out_shape,
      const TensorListShape<tensor_ndim> &in_shape,
      span<const TensorShape<ndim>> output_sizes) {
    TensorListShape<tensor_ndim> shape;
    shape.resize(in_shape.num_samples(), tensor_ndim);
    for (int i = 0; i < in_shape.num_samples(); i++) {
      auto out_tshape = out_shape[i];
      TensorShape<tensor_ndim> expected_shape;

      int in_d = 0;
      for (int j = 0; j < tensor_ndim; j++) {
        expected_shape[j] = (j == channel_dim)
         ? in_shape.tensor_shape_span(i)[channel_dim]
         : output_sizes[i][in_d++];
      }

      DALI_ENFORCE(out_tshape == expected_shape,
        "Invalid output tensor shape for sample: " + std::to_string(i));
    }
  }

  dim3 BlockDim() const {
    return dim3(block_dim_.x, block_dim_.y, block_dim_.z);