parameters: (self, axis)
docstring: Returns the size of slices across the specified dimension.
docstring_summary: Returns the size of slices across the specified dimension.
docstring_tokens: Returns the size of slices across the specified dimension .
function: def dimension_size(self, axis):
      """Returns the size of slices across the specified dimension."""
      if not isinstance(axis, int):
        raise TypeError('axis must be an integer')
      partitioned_ndims = len(self._partitioned_dim_sizes)
      if axis < partitioned_ndims:
        return self._partitioned_dim_sizes[axis]
      else:
        return self._inner_dim_sizes[axis - partitioned_ndims]
function_tokens: def dimension_size ( self , axis ) : if not isinstance ( axis , int ) : raise TypeError ( 'axis must be an integer' ) partitioned_ndims = len ( self . _partitioned_dim_sizes ) if axis < partitioned_ndims : return self . _partitioned_dim_sizes [ axis ] else : return self . _inner_dim_sizes [ axis - partitioned_ndims ]