pytorch张量转换-张量的张量答案

【问题标题】：pytorch tensor conversion - tensor of tensorspytorch张量转换-张量的张量
【发布时间】：2021-10-11 04:42:34
【问题描述】：

我在一个数据集中有两个张量，如下所示

如何转换成张量的张量？得到最终输出如下：

我尝试转换如下：

for i in (transformed_dataset):
sample = i['image'], i['label'].view(1)

我正在获取元组

for i in transformed_dataset:
sample = torch.cat((i['image'], i['label'].view(1)), dim=1)

由于不同数量的暗淡（如上的 2 和 1），猫也可以工作，我无法将第一个火炬转换为 1 暗淡，因为它的 224x224 =50176

【问题讨论】：

【解决方案1】：

解决者：

samplelist = []
for i in transformed_dataset:
    sample = i['image'], i['label'].view(1)
    samplelist.append(sample)

【讨论】：