我有一批来自 GRU 的输出隐藏向量。它的形状是[1,4,256]
( 0 ,.,.) =
-0.9944 1.0000 0.0000 ... -1.0000 0.0000 -1.0000
-1.0000 1.0000 0.0000 ... -1.0000 0.0000 -1.0000
-1.0000 1.0000 0.0000 ... -1.0000 0.0000 -1.0000
-1.0000 1.0000 0.0000 ... -1.0000 0.0000 -1.0000
[torch.cuda.FloatTensor of size (1,4,256) (GPU 0)]
我需要 [1,1,256] 的形状传递给另一个模型。我怎么能接受呢?通过这条线我只能有 [1,256] 的形状
decoder_hidden = encoder_hidden[:, index]
调整大小并包装新的 FloatTensor 无效。
最佳答案
您可以使用 view()
张量的方法。
decoder_hidden_new = decoder_hidden.view((1, 1, 256))
关于python - 如何从 [1,4,256] cuda.FloatTensor 形成 [1,1,256]?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49334900/