Warning: This project is deprecated. Swift for TensorFlow was an experiment in the next-generation platform for machine learning, incorporating the latest research across machine learning, compilers, differentiable programming, systems design, and beyond. It was archived in February 2021.

BidirectionalRecurrentLayer

public struct BidirectionalRecurrentLayer<Cell: RecurrentLayerCell>: Layer
where Cell.TimeStepOutput: Mergeable


                    
                    
                    Input

Declaration

public typealias Input = [Cell.TimeStepInput]


                    
                    
                    Output

Declaration

public typealias Output = [Cell.TimeStepOutput]


                    
                    
                    MergeFunction

Declaration

public typealias MergeFunction = @differentiable (Cell.TimeStepOutput, Cell.TimeStepOutput) -> Cell.TimeStepOutput

forward
The forward recurrent layer.
Declaration
public var forward: RecurrentLayer<Cell>
backward
The backward recurrent layer.
Declaration
public var backward: RecurrentLayer<Cell>
mergeFunction
The differentiable function used for merging forward and backward recurrent layer outputs.
Declaration
@noDerivative public var mergeFunction: MergeFunction { get }


                    
                    
                    init(_:mergeFunction:)

Creates an instance from the given recurrent layer cell and merge function.

Declaration

public init(_ cell: @autoclosure () -> Cell, mergeFunction: @escaping MergeFunction = concatenate)


                    
                    
                    callAsFunction(_:initialForwardLayerState:initialBackwardLayerState:)

Declaration

@differentiable
public func callAsFunction(
  _ inputs: Input,
  initialForwardLayerState: Cell.State,
  initialBackwardLayerState: Cell.State
) -> Output


                    
                    
                    callAsFunction(_:)

Declaration

@differentiable
public func callAsFunction(_ inputs: Input) -> Output


                    
                    
                    lastOutput(from:initialForwardLayerState:initialBackwardLayerState:)

Declaration

@differentiable
public func lastOutput(
  from inputs: Input,
  initialForwardLayerState: Cell.State,
  initialBackwardLayerState: Cell.State
) -> Cell.TimeStepOutput


                    
                    
                    lastOutput(from:)

Declaration

@differentiable
public func lastOutput(from inputs: Input) -> Cell.TimeStepOutput