torch stack理解

torch.stack的理解 官方的torch.stack說明文檔： 1、進行stack的tensor的維度必須一致，即All tensors need to be of the same size； 2、dim的取值範圍：以二維tensor爲例，dim的取值範圍爲[0,1,2] 以具體例子說明stack的用法 a,b,c都是2*3的二維tensor，dim有三種取值情況，分別是dim=0，dim

>>阅读原文<<