假設我有四個拜托奇張量。每個張量都是形狀的 。第一個維度是張量的數量,第二個維度是顏色的數量(例如,在我們的示例中是灰度),最后兩個維度表示圖像的高度和寬度。(tensor1, tensor2, tensor3, tensor4)(1, 1, 84, 84)我想堆疊它們,以便我得到形狀。(4, 1, 84, 84)我試過了,但我得到了一個形狀。torch.stack((tensor1, tensor2, tensor3, tensor4), dim=0)(4, 1, 1, 84, 84)如何堆疊這些張量,使形狀(4, 1, 84, 84)
用藥劑獲得形狀(4,1,84,84)
慕蓋茨4494581
2022-09-27 09:33:49