public struct Function<Input, Output> : ParameterlessLayer where Input : Differentiable, Output : Differentiable
Uma camada que inclui uma função diferenciável personalizada.
Declaração
public typealias TangentVector = EmptyTangentVector
Declaração
public typealias Body = @differentiable (Input) -> Output
Declaração
@noDerivative public let body: Body
Declaração
public init(_ body: @escaping Body)
Declaração
@differentiable public func callAsFunction(_ input: Input) -> Output