2024-11-13 厉飞雨 阅读(4) 评论(0) 赞(0)

张量的拼接操作在神经网络搭建过程中是非常常用的方法，例如: 在残差网络、注意力机制中都使用到了张量拼接。 1. torch.cat 函数的使用 2. torch.stack 函数的使用 1. torch.cat 函数的使用 {#title-0} ============================= torch.cat 函数主要用于根据指定的维度将两个张量拼接到一起，...