for X_batch, y_batch in dataloader ，其中y_batch想转成list
for X_batch, y_batch in dataloader ，其中y_batch想转成listfor X_batch, y_batch in dataloader:
# y_batch数据类型的是 tensor组成的list
# 初始化一个空列表来存储展平后的数据
y_batch_flat = []
# 遍历y_batch 这个list中的每个张量
for tensor in y_batch:
# 展平张量并添加到列表中
# 这里假设每个张量都只包含一个元素，使用.item()转换为Python标量
# 如果张量包含多个元素，使用tensor.flatten()或tensor.view(-1)展平张量
y_batch_flat.extend(tensor.flatten().tolist())# 或者使用
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。
页:
[1]