lamp.nn.graph.GraphAttention
See theGraphAttention companion object
case class GraphAttention(wNodeKey1: Constant, wNodeKey2: Constant, wEdgeKey: Constant, wNodeValue: Constant, wAttention: Option[Constant], nonLinearity: Boolean, dropout: Dropout, numHeads: Int) extends GenericModule[Graph, Graph]
Attributes
- Companion
- object
- Graph
-
- Supertypes
-
trait Serializabletrait Producttrait Equalsclass Objecttrait Matchableclass AnyShow all
Members list
Value members
Concrete methods
The implementation of the function.
The implementation of the function.
In addition of x
it can also use all the `state to compute its value.
Attributes
- Definition Classes
Inherited methods
Alias of forward
Computes the gradient of loss with respect to the parameters.
Computes the gradient of loss with respect to the parameters.
Attributes
- Inherited from:
- GenericModule
Returns the total number of optimizable parameters.
Returns the state variables which need gradient computation.
Returns the state variables which need gradient computation.
Attributes
- Inherited from:
- GenericModule
Attributes
- Inherited from:
- Product
Attributes
- Inherited from:
- Product
Attributes
- Inherited from:
- GenericModule
In this article