vis4d.op.util

Utilities for op.

Functions

unmap(data, count, inds[, fill])

Unmap a subset of data back to the original data (of size count).
unmap(data, count, inds, fill=0)[source]

Unmap a subset of data back to the original data (of size count).

Parameters:

  • data (Tensor) – Subset of the original data.

  • count (int) – Length of the original data.

  • inds (Tensor) – Indices of the subset entries in the original set.

  • fill (int, optional) – Fill value for other entries. Defaults to 0.

Returns:

Tensor sized like original data that contains the subset.

Return type:

Tensor