x = x.view()含义

x = x.view(batch_size * seq_len, x.size(2), x.size(3))
转换大小,四维变为三维

x = x.view(x.size(0), -1)
view()函数的功能与reshape类似,用来转换size大小。x = x.view(batchsize, -1)中batchsize指转换后有几行,而-1指在不告诉函数有多少列的情况下,根据原tensor数据和batchsize自动分配列数。

你可能感兴趣的:(pytorch,python,人工智能)