| 395 | } |
| 396 | |
| 397 | daliResult_t daliTensorListGetShape( |
| 398 | daliTensorList_h tensor, |
| 399 | int *out_num_samples, |
| 400 | int *out_ndim, |
| 401 | const int64_t **out_shape) { |
| 402 | DALI_PROLOG(); |
| 403 | auto *ptr = ToPointer(tensor); |
| 404 | auto &shape = ptr->GetShape(); |
| 405 | if (out_ndim) |
| 406 | *out_ndim = shape.sample_dim(); |
| 407 | if (out_num_samples) |
| 408 | *out_num_samples = shape.num_samples(); |
| 409 | if (out_shape) |
| 410 | *out_shape = shape.shapes.data(); |
| 411 | DALI_EPILOG(); |
| 412 | } |
| 413 | |
| 414 | daliResult_t daliTensorListGetByteSize( |
| 415 | daliTensorList_h tensor_list, |