У меня есть вход X формы:
(batch_size, no_features, no_nodes, timestamps)
и матрица смежности A размером
(no_nodes, no_nodes).
Когда я хочу свернуть X, AI может просто использовать torch.einsum('ncvl,vw->ncwl', (X, A)).
Теперь возникает мой вопрос: если вместо одной матрицы у меня будет одна для каждой временной метки с формой
(batch_size, timestamps, no_nodes, no_nodes)
, как бы я свертывал каждый сигнал X с соответствующей матрицей?(партия будет соответствовать, и отметка времени).