Searched refs:arr_shape (Results 1 – 1 of 1) sorted by relevance
327 def _move_axes_for_concatenate(arr_shape, axis): argument333 original_axes = tuple(range(len(arr_shape)))335 new_shape = arr_shape[1:axis+1] + (arr_shape[0] * arr_shape[axis+1],) + \336 arr_shape[axis+2:]406 arr_shape = F.shape(arrays)407 _check_axes_range((axis,), len(arr_shape))410 new_axes, new_shape = _move_axes_for_concatenate(arr_shape, axis)426 arr_shape = F.shape(arrays[0])427 _check_axes_range((axis,), len(arr_shape))1422 arr_shape = x.shape[all …]