记录一些关于张量的操作,会持续更新
flatten()
用于展平张量的操作,目前我见到的有两个函数,分别是:
torch.flatten()
torch.Tensor.flatten()
二者其实是几乎一样的。以前者为例,函数原型为torch.flatten(input, start_dim=0,end_dim=-1)
。其中start_dim
是展平开始的维度,end_dim
是展平结束的维度。例如,对于一个3x2x2
的张量a
,torch.flatten(a, start_dim=0,end_dim=-1)
,会返回一个一维数组,torch.flatten(a, start_dim=0,end_dim=1)
返回一个6x2
的数组,torch.flatten(a, start_dim=1,end_dim=-1)
返回一个3x4
的数组。