lamp-core/lamp.nn.graph/MPNN

MPNN

object MPNN
Companion:
class
trait Product
trait Mirror
class Object
trait Matchable
class Any
MPNN.type

Type members

Inherited types

type MirroredElemLabels <: Tuple

The names of the product elements

The names of the product elements

Inherited from:
Mirror
type MirroredLabel <: String

The name of the type

The name of the type

Inherited from:
Mirror

Value members

Concrete methods

def aggregate[S : Sc](numVertices: Long, message: Variable, edgeI: STen, edgeJ: STen, degreeNormalizeI: Boolean, degreeNormalizeJ: Boolean, aggregateJ: Boolean): Variable
def countOccurences(t: STen, elems: Long)(implicit scope: Scope): STen

Implicits

Implicits

implicit def load[M1 <: Module : Load, M2 <: Module : Load]: Load[MPNN[M1, M2]]
implicit def trainingMode[M1 <: Module : TrainingMode, M2 <: Module : TrainingMode]: TrainingMode[MPNN[M1, M2]]