PyTorch的TensorDataset功能怎么使用

云技术五哥 • 05/14 21:03 • 开发 • 58 views

本文小编为大家详细介绍“PyTorch的TensorDataset功能怎么使用”，内容详细，步骤清晰，细节处理妥当，希望这篇“PyTorch的TensorDataset功能怎么使用”文章能帮助大家解决疑惑，下面跟着小编的思路慢慢深入，一起来学习新知识吧。TensorDataset 可以用来对tensor进行打包，就好像 python 中的 zip 功能。该类通过每一个 tensor 的第一个维度进行索引。因此，该类中的 tensor 第一维度必须相等。运行结果：(tensor([[1, 2, 3],
[4, 5, 6]]), tensor([44, 55]))
================================================================================
tensor([1, 2, 3]) tensor(44)
tensor([4, 5, 6]) tensor(55)
tensor([7, 8, 9]) tensor(66)
tensor([1, 2, 3]) tensor(44)
tensor([4, 5, 6]) tensor(55)
tensor([7, 8, 9]) tensor(66)
tensor([1, 2, 3]) tensor(44)
tensor([4, 5, 6]) tensor(55)
tensor([7, 8, 9]) tensor(66)
tensor([1, 2, 3]) tensor(44)
tensor([4, 5, 6]) tensor(55)
tensor([7, 8, 9]) tensor(66)
================================================================================
batch:1 x_data:tensor([[1, 2, 3],
[1, 2, 3],
[4, 5, 6],
[4, 5, 6]]) label: tensor([44, 44, 55, 55])
batch:2 x_data:tensor([[4, 5, 6],
[7, 8, 9],
[7, 8, 9],
[7, 8, 9]]) label: tensor([55, 66, 66, 66])
batch:3 x_data:tensor([[1, 2, 3],
[1, 2, 3],
[7, 8, 9],
[4, 5, 6]]) label: tensor([44, 44, 66, 55])注意：TensorDataset免费云主机域名中的参数必须是 tensorPytorch中，TensorDataset()可以快速构建训练所用的数据，不用使用自建的Mydataset()，如果没有熟悉适用的dataset可以使用TensorDataset()作为暂时替代。只需要把data和label作为参数输入，就可以快速构建，之后便可以用Dataloader处理。读到这里，这篇“PyTorch的TensorDataset功能怎么使用”文章已经介绍完毕，想要掌握这篇文章的知识点还需要大家自己动手实践使用过才能领会，如果想了解更多相关内容的文章，欢迎关注百云主机行业资讯频道。