Struct tguf::Batch
Container for temporal edge data.
#include <tguf.h>
Public Attributes
| Type | Name |
|---|---|
| torch::Tensor | dst Destination node IDs [B]. |
| torch::Tensor | msg Edge features [B, msg_dim]. |
| std::optional< torch::Tensor > | neg_dst |
| torch::Tensor | src Source node IDs [B]. |
| torch::Tensor | time Timestamps [B]. |
Public Attributes Documentation
variable dst
Destination node IDs [B].
torch::Tensor tguf::Batch::dst;
variable msg
Edge features [B, msg_dim].
torch::Tensor tguf::Batch::msg;
variable neg_dst
std::optional<torch::Tensor> tguf::Batch::neg_dst;
Optional negative destinations for link prediction [B, negatives_per_edge]
variable src
Source node IDs [B].
torch::Tensor tguf::Batch::src;
variable time
Timestamps [B].
torch::Tensor tguf::Batch::time;
The documentation for this class was generated from the following file include/tguf.h