fsp_matrix

FSP matrix op

fsp_matrix op用于计算两个4-D Tensor特征图的求解过程（FSP）矩阵。假设特征图x的形状为

，特征图y的形状为

1.将x reshape到形状为

的矩阵，将y reshape到形状为

的矩阵。

输出是一个batch的fsp矩阵。

参数：

返回：一个形状为 [batch_size,x_channel,y_channel][batch_size,x_channel,y_channel] 的fsp矩阵, 是一个 3-D Tensor，数据类型与输入数据类型一致。其中，x_channel是输入x的通道数，y_channel是输入y的通道数。数据类型为float32或float64。

代码示例