gather
fun DTensor.gather(indices: List<Int>, axis: Int, paddingIndex: Int = -1): DTensor
Content copied to clipboard
Return a tensor made up of slices taken from the input tensor at the given indices along the given axis.
Return
result tensor
Parameters
indices
indices to gather slices from
axis
axis long which to slice x
fun FloatTensor.gather(indices: List<Int>, axis: Int, paddingIndex: Int = -1): FloatTensor
Content copied to clipboard