相关疑难解决方法(0)

我正在创建一个卷积稀疏自动编码器,我需要将一个充满值(其形状[samples, N, N, D])的4D矩阵转换为稀疏矩阵.

对于每个样本,我有D NxN特征映射.我想将每个NxN特征映射转换为稀疏矩阵,最大值映射为1,其他所有映射为0.

我不想在运行时这样做但在Graph声明期间(因为我需要使用生成的稀疏矩阵作为其他图形操作的输入),但我不明白如何获得索引来构建稀疏矩阵.