tensor和lable转化成DataLoader
import torch
from torch.utils.data import TensorDataset, DataLoader
# 假设有一些张量和对应的标签
tensors = [torch.randn(32, 10) for _ in range(100)] # 示例张量列表
labels = torch.tensor([0, 1] * 50) # 示例标签
# 创建TensorDataset
dataset = TensorDataset(tensors, labels)
# 创建DataLoader
batch_size = 16
train_dataloader = DataLoader(dataset, batch_size=batch_size, shuffle=True)
# 使用dataloader
for batch_index, (data, target) in enumerate(train_dataloader):
# 在这里处理每个批次的数据和标签
# 例如,可以将数据送入模型进行训练
print(f"Batch Index: {batch_index}, Data Size: {data.size()}, Label Size: {target.size()}")
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· Ollama——大语言模型本地部署的极速利器
· 使用C#创建一个MCP客户端
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· ollama系列1:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
2020-04-23 问题 B: 寻路