基于时间序列合并两个张量

Question

我有两个张量：

A -> (128,19,3,99,99) #(batch, date, data, data, data)
B -> (128,9,223)  #(batch, date, data)

编码后我会有类似的东西：

A2 -> (128,19,10) # (batch, date, encoded data with CNN_1)
B2 -> (128,9,10) # (batch, date, encoded data with CNN_2)
C -> (128,28,10) # merge of A2 and B2

我有每个张量(A,B)的时间序列记录，我希望能够将它们在date轴上合并为一个张量。

以下是张量 A 的时间序列示例： ['2021-12-13', '2021-12-28', '2022-01-02', '2022-01-10', '2022-01-20', '2022-01-26', '2022-02-06', '2022-02-14', '2022-02-22', '2022-03-02', '2022-03-17', '2022-03-21', '2022-03-27', '2022-03-30', ...]张量 B 具有不同的日期，不一定落在张量 A 的确切日期上。

所以最后，张量C中的date轴将是这样的： A2,A2,B2,A2,B2,A2,B2,B2取决于日期的原始顺序。

任何帮助都会很棒！

Answer 1

我得到了答案：）

import torch

A0_RGB = torch.rand((128,19,3))
A1_Hyper = torch.rand((128,9,3))

A2_concat = torch.concat((A0_RGB,A1_Hyper), dim=1)
A3_sorted_v, A3_sorted_i = A2_concat[0,:,0].sort()
A5_test_selection = A2_concat[:,A3_sorted_i,:]

只是概念...