LSTM Class Reference


                    
                    
                    init(masked:hiddenSize:numberOfLayers:projectSize:bias:batchFirst:bidirectional:dropout:trainable:name:)

Declaration

Swift

public init(
  masked: Bool, hiddenSize: Int, numberOfLayers: Int, projectSize: Int? = nil, bias: Bool = true,
  batchFirst: Bool = true, bidirectional: Bool = false, dropout: Float? = nil,
  trainable: Bool? = nil, name: String = ""
)


                    
                    
                    callAsFunction(_:mask:streamContext:)

Declaration

Swift

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


                    
                    
                    callAsFunction(_:streamContext:)

Declaration

Swift

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