随笔分类 - pytorch
摘要:作用 collate_fn:即用于collate的function,用于整理数据的函数。 说到整理数据,你当然要会用数据,即会用数据制作工具torch.utils.data.Dataset,虽然我们今天谈的是torch.utils.data.DataLoader。 collate_fn笼统的说就是用
阅读全文
摘要:1、 squeeze()函数 去除size为1的维度,包括行和列 至于维度大于等于2时,squeeze()不起作用 torch.unsqueeze(A,N)函数的作用减少数组A指定位置N的维度,如果指定的维度大于1,那么将操作无效,如果不指定维度N,那么将删除所有维度为1的维度 行变的例子 >>>
阅读全文