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

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

* (張量張量的元組,名稱 dim,張量輸出)

* (張量張量的元組,名稱 dim,張量輸出)

慕的地6264312 2022-09-27 09:51:59
假設我有內存列表。每個元素都是形狀的pytorch張量。list_of_tensors = [tensor1, tensor2, tensor3, tensor4](1, 1, 84, 84)我想連接張量列表以獲得形狀的張量。 也許肯定會允許我這樣做。 必須是張量的元組,所以或不起作用。(4, 1, 84, 84)torch.cat(TT, dim=0)TTtorch.cat(*list_of_tensors, dim=0)torch.cat((*list_of_tensors), dim=0)如何使用和創建新的形狀張量list_of_tensorstorch.cat(???, dim=0)(4, 1, 84, 84)
查看完整描述

1 回答

?
慕桂英546537

TA貢獻1848條經驗 獲得超10個贊

您可以使用堆棧,并通過擠壓去除多余的尺寸

c = (torch.stack(list_of_tensors,dim=1)).squeeze(0)

現在 c.形狀是 (4, 1, 84, 84)

你可以在這里找到解釋:https://discuss.pytorch.org/t/how-to-turn-a-list-of-tensor-to-tensor/8868/6


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

添加回答

舉報

0/150
提交
取消
微信客服

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

幫助反饋 APP下載

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

公眾號

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