mindspore.ops.combinations

mindspore.ops.combinations(x, r=2, with_replacement=False)[source]

Returns all r-length subsequences of input Tensor.

When with_replacement is set to False, it works similar to Python’s itertools.combinations, and when with_replacement is set to True, it behaves like itertools.combinations_with_replacement.

Parameters

x (Tensor) – One-dimensional tensors.
r (int, optional) – Number of elements to perform combination. Default: 2.
with_replacement (bool, optional) – Allow duplication or not. Default: False.

Returns

Tensor, contains all possible combinations of elements sampled from input Tensor.

Raises

TypeError – If x is not a tensor.
TypeError – If x is not an int.
TypeError – If with_replacement is not bool.
ValueError – If x is not one-dimensional.

Supported Platforms:: Ascend GPU CPU

Examples

>>> x = Tensor([1, 3, -1, 0, 4])
>>> output = ops.combinations(x)
>>> print(output.asnumpy())
[[ 1  3]
 [ 1 -1]
 [ 1  0]
 [ 1  4]
 [ 3 -1]
 [ 3  0]
 [ 3  4]
 [-1  0]
 [-1  4]
 [ 0  4]]