Returns a random permutation of integers from 0 to n - 1.

tch_randperm(n, dtype = NULL, layout = NULL, device = NULL,
  requires_grad = FALSE)

Arguments

n

the upper bound (exclusive).

dtype

the desired data type of returned tensor. Default: if NULL, infers data type from x.

device

the desired device of returned tensor. Default: if NULL, uses the current device for the default tensor type (see tch_set_default_tensor_type()). device will be the CPU for CPU tensor types and the current CUDA device for CUDA tensor types.

requires_grad

If autograd should record operations on the returned tensor. Default: FALSE.

the

desired layout of returned Tensor. Default: 'strided'

Details

The shape of the tensor is defined by the variable argument sizes.

Examples

tch_randperm(4)
#> tensor #> 0 #> 1 #> 2 #> 3 #> [ Variable[CPUFloatType]{4} ]