9i_ SSKrSSKJr SSKJr SSjr\RS5r\RSS\R4Sjj5r \RSS\R4Sjj5r g) N)TensorDescriptorc[U5n[UR5n[U5nUS:aX$- nSUs=::aUS- :dS5e S5eUS::dS5e[U5U:XdS5eSnSnX2U::dS 5eXcU'URU5nS U- U/[ U5Vs/sHoRU5PM sn-n XU/U-n SS/U-n [ X X5$s snf) a Given a 2- or 3-dimensional tensor T, this creates a 'ragged descriptor' which behaves like a concatenation (along the first axis) of subarrays of potentially unequal size. The load_ragged and store_ragged device functions can be used to read and write from subarrays T[batch_offset : batch_offset + batch_size] with hardware bounds-checking preventing any sort of leakage outside the subarray. rzlast dimension cannot be raggedz?r:sv ; %AP   2<< " U",, U  Ur#