Searched refs:_unravel_index (Results 1 – 1 of 1) sorted by relevance
1945 res_tensor = _unravel_index(indices, shape)1949 def _unravel_index(indices: Tensor, shape: Union[int, Sequence[int]]) -> Tensor: function