我的输入X的形状为:
(batch_size, no_features, no_nodes, timestamps)
和大小为
的邻接矩阵A(no_nodes, no_nodes).
当我想对X和A进行卷积时,我可以简单地使用torch.einsum('ncvl,vw->ncwl', (X, A))
。
现在我的问题来了:如果不是每个矩阵,每个时间戳都有一个矩阵
(batch_size, timestamps, no_nodes, no_nodes)
如何将每个信号X与对应的矩阵进行卷积? (批处理将与时间戳匹配)。