Method append

include/dali/core/tensor_shape.h:716–743

@{ */ * Append this TensorListShape with shapes from other TensorListShapes. */

   * Append this TensorListShape with shapes from other TensorListShapes.
   */
  void append(span<const TensorListShape<sample_ndim>> tlss) {
    int new_samples = 0;
    for (const auto &tls : tlss) {
      new_samples += tls.num_samples();
    }
    if (new_samples == 0) return;

    // The following line is imperfect, because uninitialized TensorListShape<-1>
    // has `ndim` == 0, while `0` is legal number of dimensions.
    bool is_uninitialized = empty() && sample_dim() <= 0;

    auto reference_dim_value = is_uninitialized ? tlss[0].sample_dim() : sample_dim();
    for (const auto &tls : tlss) {
      if (tls.sample_dim() != reference_dim_value) {
        throw std::invalid_argument(
                "All samples in TensorListShape must have the same number of dimensions");
      }
    }
    nsamples += new_samples;
    if (is_uninitialized) {
      // If `this` TLS has DynamicDimensions and wasn't initialized yet
      set_sample_dim(reference_dim_value);
    }
    shapes.reserve(nsamples * sample_dim());
    for (const auto &tls : tlss) {
      shapes.insert(shapes.end(), tls.shapes.begin(), tls.shapes.end());
    }
  }

  void append(const TensorListShape<sample_ndim> &tls) {
    append(make_cspan(&tls, 1));