jittor_geometric.nn.dense

Dense layers used in Dynamic Graph Neural Networks.

class jittor_geometric.nn.dense.MergeLayer(input_dim1, input_dim2, hidden_dim, output_dim)

Bases: Module

execute(input_1, input_2)

merge and project the inputs :type input_1: Var :param input_1: Var, shape (, input_dim1) :type input_2: :sphinx_autodoc_typehints_type:`:py:class:`~jittor.jittor_core.Var`` :param input_2: Var, shape (, input_dim2) :return:

class jittor_geometric.nn.dense.TimeEncoder(time_dim, parameter_requires_grad=True)

Bases: Module

execute(timestamps)

compute time encodings of time in timestamps :type timestamps: Var :param timestamps: Var, shape (batch_size, seq_len) :return: