package graph
Linear Supertypes
Ordering
- Alphabetic
- By Inheritance
Inherited
- graph
- AnyRef
- Any
- Hide All
- Show All
Visibility
- Public
- Protected
Type Members
- case class GCN[M <: Module](transform: M with Module) extends GraphModule with Product with Serializable
- case class Graph(nodeFeatures: Variable, edgeFeatures: Variable, edgeI: STen, edgeJ: STen, vertexPoolingIndices: STen) extends Product with Serializable
- case class GraphAttention(wNodeKey1: Constant, wNodeKey2: Constant, wEdgeKey: Constant, wNodeValue: Constant, wAttention: Option[Constant], nonLinearity: Boolean, dropout: Dropout, numHeads: Int) extends GenericModule[Graph, Graph] with Product with Serializable
- type GraphModule = GenericModule[Graph, Graph]
- case class MPNN[M1 <: Module, M2 <: Module](messageTransform: M1 with Module, vertexTransform: M2 with Module, degreeNormalizeI: Boolean = true, degreeNormalizeJ: Boolean = true, aggregateJ: Boolean = true) extends GraphModule with Product with Serializable
Value Members
- object GCN extends Serializable
- object GraphAttention extends Serializable
- object MPNN extends Serializable
- object VertexPooling