for X_batch, y_batch in dataloader ，其中y_batch想转成list

for X_batch, y_batch in dataloader ，其中y_batch想转成list

1. for X_batch, y_batch in dataloader:
2.     # y_batch数据类型的是 tensor组成的list
3.     # 初始化一个空列表来存储展平后的数据
4.     y_batch_flat = []
6.     # 遍历y_batch 这个list中的每个张量
7.     for tensor in y_batch:
8.         # 展平张量并添加到列表中
9.         # 这里假设每个张量都只包含一个元素，使用.item()转换为Python标量
10.         # 如果张量包含多个元素，使用tensor.flatten()或tensor.view(-1)展平张量
11.         y_batch_flat.extend(tensor.flatten().tolist())  # 或者使用

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。