[PyTorch] 拼接多個tensor:torch.cat((A,B),axis)

注:參考博客Pytorch中的torch.cat()函數。本人在其基礎上增長了更爲詳細的解釋。html torch.cat((A,B),axis)是對A, B兩個tensor進行拼接。web 參數axis指定拼接的方式。axis=0爲按行拼接;axis=1爲按列拼接。svg 拼接的時候把待拼接的tensor視做總體。(注意在示例中理解這句話)函數 import torch # 初始化三個 ten
相關文章
相關標籤/搜索