IndexSelect

Swift

public init(name: String = "")

Swift

public func callAsFunction<T: DynamicGraph.TensorGroup, U: DynamicGraph.TensorGroup>(
  _ x: T, index: U, streamContext: StreamContext? = nil
) -> T where U.ElementNumeric == Int32, T.AnyTensor == U.AnyTensor

Swift

public func callAsFunction<T: DynamicGraph.TensorGroup, U: DynamicGraph.TensorGroup>(
  _ x: T, index: U, streamContext: StreamContext? = nil
) -> T where U.ElementNumeric == Float32, T.AnyTensor == U.AnyTensor