| 209 | } |
| 210 | |
| 211 | daliResult_t daliTensorGetShape( |
| 212 | daliTensor_h tensor, |
| 213 | int *out_ndim, |
| 214 | const int64_t **out_shape) { |
| 215 | DALI_PROLOG(); |
| 216 | auto *ptr = ToPointer(tensor); |
| 217 | auto &shape = ptr->GetShape(); |
| 218 | if (out_ndim) |
| 219 | *out_ndim = shape.sample_dim(); |
| 220 | if (out_shape) |
| 221 | *out_shape = shape.data(); |
| 222 | DALI_EPILOG(); |
| 223 | } |
| 224 | |
| 225 | daliResult_t daliTensorGetByteSize( |
| 226 | daliTensor_h tensor, |