shape

fun shape(elementSize: Int): List<Long>

Returns 2-dimensional shape of the data batch.