jactorch.functional.range#

Numeric range functions.

Functions

meshgrid(input1[, input2, dim])

Perform np.meshgrid along given axis.

meshgrid_exclude_self(input[, dim])

Exclude self from the grid.

Functions

meshgrid(input1, input2=None, dim=-1)[source]#

Perform np.meshgrid along given axis. It will generate a new dimension after dim.

meshgrid_exclude_self(input, dim=1)[source]#

Exclude self from the grid. Specifically, given an array a[i, j] of n * n, it produces a new array with size n * (n - 1) where only a[i, j] (i != j) is preserved.

The operation is performed over dim and dim +1 axes.