Pytorch中的Conv1d()函数

Posted 2023-04-25

参考技术A 举例：实体链接（x，y），x，y是两个实体，当x,y为同一实体，标注为1，否则标注为1。

这里32为batch_size，8为实体中词的个数，200为词向量。50为卷积核的数量，2为卷积核的尺寸。
输入一维卷积的时候，需要将32*8*200变换为32*200*8，因为一维卷积是在最后维度上扫的，卷积核大小为200*2，最后out的大小即为：32*50*（8-2+1）=32*50*7，最大池化后的大小为：32*50*1

pytorch中gather函数的理解。

参考技术A 函数 torch.gather(input, dim, index, out=None) → Tensor
沿给定轴 dim ,将输入索引张量 index 指定位置的值进行聚合.
对一个 3 维张量,输出可以定义为:

Parameters:

使用说明举例：