torch.stack():函数stack()
对序列数据内部的张量进行扩维拼接,指定维度由程序员选择(对 原始张量按照指定维度进行拼接,拼接后维度为原始维度加一,如原始维度为2*2,那么拼接后为n*2*2或2*n*2或2*2*n)。
详情:https://blog.csdn.net/xinjieyuan/article/details/105205326
torch.cat(): 在给定维度上对输入的张量序列seq 进行连接操作。(连接后维度为原始维度,如原来为2*2,那么连接后为n*2或2*n)
详情:https://blog.csdn.net/xinjieyuan/article/details/105208352