摘要:
1. view( )函数 PyTorch 中的view( )函数相当于numpy中的resize( )函数,都是用来重构(或者调整)张量维度的,用法稍有不同。 >>> import torch >>> re = torch.tensor([1, 2, 3, 4, 5, 6]) >>> result 阅读全文
摘要:
TensorDataset TensorDataset可以用来对 tensor 进行打包,就好像 python 中的 zip 功能。该类通过每一个 tensor 的第一个维度进行索引。因此,该类中的 tensor 第一维度必须相等. 另外:TensorDataset 中的参数必须是 tensor i 阅读全文