亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

用藥劑獲得形狀(4,1,84,84)

用藥劑獲得形狀(4,1,84,84)

慕蓋茨4494581 2022-09-27 09:33:49
假設我有四個拜托奇張量。每個張量都是形狀的 。第一個維度是張量的數量,第二個維度是顏色的數量(例如,在我們的示例中是灰度),最后兩個維度表示圖像的高度和寬度。(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)
查看完整描述

1 回答

?
蠱毒傳說

TA貢獻1895條經驗 獲得超3個贊

您可以使用串聯函數:

a = torch.ones(1,1,84,84)
b = torch.ones(1,1,84,84)
c = torch.cat((a,b), 0) # size[2,1,84,84]


查看完整回答
反對 回復 2022-09-27
  • 1 回答
  • 0 關注
  • 93 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號