torch.cat

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了torch.cat相关的知识,希望对你有一定的参考价值。

参考技术A

torch.cat 的作用是把两个 tensor 合并为一个 tensor
第一个参数是需要连接的tensor list , 第二个参数指定按照哪个维度进行拼接

按照列进行拼接

这个函数的应用是基础的,很多任务,例如要处理的数据是一行一行的,分别都转换为 tensor 之后,需要将全部的句子都拼接起来,然后再分成 batch 批量输入模型,所以需要用到 cat 的操作;

torch.cat拼接

以上是关于torch.cat的主要内容,如果未能解决你的问题,请参考以下文章