Thanks for tuning in to Google I/O. View all sessions on demand
Watch on demand
Module
Stay organized with collections
Save and categorize content based on your preferences.
-
The input type of the layer.
-
The output type of the layer.
-
Returns the output obtained from applying the layer to the given input.
Declaration
@differentiable(wrt: self)
func callAsFunction(_ input: Input) -> Output
-
Returns the output obtained from applying the layer to the given input.
Default Implementation
Returns the output obtained from applying the layer to the given input.
Declaration
@differentiable(wrt: self)
func forward(_ input: Input) -> Output
-
Default Implementation
Returns the annotated output obtained from applying the layer to the
given input.
Declaration
@differentiable(wrt: self)
public func callAsFunction(_ input: Input) -> Output
Return Value
The annotated output.
-
Annotates output
.
Note: Returns output
if using a backend that does not support annotations.
Declaration
@differentiable
public func annotated(_ output: Output) -> Output
Return Value
The annotated output.
-
Returns the annotations obtained from applying the layer to the given input.
Declaration
public func summary(input: Input) -> String
Return Value
All collected annotations from the XLA graph.
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2021-09-28 UTC.
[{
"type": "thumb-down",
"id": "missingTheInformationINeed",
"label":"Missing the information I need"
},{
"type": "thumb-down",
"id": "tooComplicatedTooManySteps",
"label":"Too complicated / too many steps"
},{
"type": "thumb-down",
"id": "outOfDate",
"label":"Out of date"
},{
"type": "thumb-down",
"id": "samplesCodeIssue",
"label":"Samples / code issue"
},{
"type": "thumb-down",
"id": "otherDown",
"label":"Other"
}]
[{
"type": "thumb-up",
"id": "easyToUnderstand",
"label":"Easy to understand"
},{
"type": "thumb-up",
"id": "solvedMyProblem",
"label":"Solved my problem"
},{
"type": "thumb-up",
"id": "otherUp",
"label":"Other"
}]