pub fn take<R, T, B, D, I>(
tensor: &TensorAny<R, T, B, D>,
indices: I,
axis: isize,
) -> Tensor<T, B, D>where
R: DataAPI<Data = <B as DeviceRawAPI<T>>::Raw>,
D: DimAPI + DimSmallerOneAPI,
D::SmallerOne: DimAPI,
B: DeviceAPI<T> + DeviceIndexSelectAPI<T, D> + DeviceCreationAnyAPI<T>,
I: TryInto<AxesIndex<isize>, Error = Error>,