temp_output1 = Kexpand_dimssupports0 -1 temp_output2 = Kexpand_dimssupports1 -1 temp_output = Kconcatenatetemp_output1 temp_output2 axis=2 temp_output = Lambdaexpend_tensor_di

temp_output1 = supports[0].unsqueeze(-1) temp_output2 = supports[1].unsqueeze(-1) temp_output = torch.cat((temp_output1, temp_output2), dim=2) temp_output = temp_output.unsqueeze(0) # expend_tensor_dim_l函数的作用是在第一维度上增加一个维度，因此这里用unsqueeze(0)实现同样的操作。