0
点赞
收藏
分享

微信扫一扫

深入理解 Promise、async、回调函数和 AJAX

SPEIKE 2023-06-26 阅读 33

函数解答:

1、data.TensorDataset可以用来对tensor进行打包。

更明白点来说就是将样本和lable合并起来(样本的个数于lable的长度必须一致)

input:连个相同长度的tensor

output:打包好的tensor

example:


如何显示打包后的tensor:

 for x_train, y_label in train_ids:  对样本和lable分开输出

next(iter(data_iter)):打包好的数据整体输出

2、 data.DataLoader(dataset, batch_size, shuffle=is_train)

对data.TensorDataset打包好的数据进行一个选取。

batch_size=选取样本的个数

shuffle=true随机选取,false顺序选取
 

举报

相关推荐

0 条评论