Method tensor_shape

include/dali/core/tensor_view.h:317–319

  template <int output_dim = sample_ndim>
  TensorShape<output_dim> tensor_shape(int sample) const {
    return shape.template tensor_shape<output_dim>(sample);
  }

  auto tensor_shape_span(int sample) {
    return shape.tensor_shape_span(sample);