卷积多个时间戳

时间:2019-07-05 10:39:46

标签: python pytorch numpy-einsum

我的输入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与对应的矩阵进行卷积? (批处理将与时间戳匹配)。

0 个答案:

没有答案