similarity_focus

    实现SimilarityFocus(相似度聚焦)运算

    通过以下三个步骤,该层生成一个和输入 input 同形的 similarity focus mask(相似度聚焦掩码):

    1. 根据 axis 和 提取一个三维张量,第一维为batch大小。 例如,如果 axis=1, indexes=[a] , 将得到矩阵 T=X[:, a, :, :] 。 该例中,如果输入X的形为 (BatchSize, A, B, C) ,则输出张量T的形为 (BatchSize, B, C) 。

    请参考 。

    参数:

    • axis (int) – 指明要选择的轴。 可能取值为 1, 2 或 3。

    返回:一个和输入 Variable 同形状、同数据类型的 Variable

    代码示例

    1. data = fluid.layers.data(