[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
相关文章
相关标签/搜索