Slice input along dim at index, removing that dimension.
Parameters
inputTensorInput tensor with at least 1 dimension.
dimintDimension to slice.
indexintIndex to select along
dim. Negative indices count from the end.Returns
TensorView of input with dimension dim removed.
Examples
>>> import lucid
>>> x = lucid.arange(6).reshape(2, 3)
>>> lucid.select(x, 1, -1)
tensor([2, 5], dtype=lucid.int64)