Embedding

Swift

public init<T: TensorNumeric>(
  _ dataType: T.Type, vocabularySize: Int, embeddingSize: Int, trainable: Bool? = nil,
  name: String = ""
)

Swift

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

Swift

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