torch (version 0.0.2)

torch_cartesian_prod: Cartesian_prod

Description

Cartesian_prod

Arguments

*tensors

NA any number of 1 dimensional tensors.

TEST

Do cartesian product of the given sequence of tensors. The behavior is similar to python's itertools.product.

Examples

# NOT RUN {
if (torch_is_installed()) {

a = c(1, 2, 3)
b = c(4, 5)
tensor_a = torch_tensor(a)
tensor_b = torch_tensor(b)
torch_cartesian_prod(list(tensor_a, tensor_b))
}
# }