编写pytorch代码，给时间集添一个标签维度（样本，长度大小，特征）

假设你的时间集是一个张量 `time_data`，形状为 `(batch_size, seq_length, feature_dim)`，其中 `batch_size` 是批量大小，`seq_length` 是时间序列长度，`feature_dim` 是每个时间步的特征维度。

你可以通过以下代码向 `time_data` 添加一个标签维度：

import torch

# 假设标签数量为 10
label_num = 10

# 将标签维度设为 1
time_data_with_label = time_data.unsqueeze(1)

# 创建标签张量，形状为 (batch_size, 1, seq_length)
label_data = torch.randint(label_num, (batch_size, 1, seq_length))

# 将标签张量与时间集张量拼接在一起，形状为 (batch_size, label_num, seq_length, feature_dim)
time_data_with_label = torch.cat([time_data_with_label, label_data], dim=1)

其中，`unsqueeze` 方法可以在指定维度上添加一个维度，`torch.randint` 方法可以生成随机整数标签张量。最后一行代码使用 `torch.cat` 方法将标签张量和时间集张量在标签维度上拼接在一起。