首页 文章
  • 1 votes
     answers
     views

    Tensorflow通过二进制矩阵计算乘法

    我有我的数据张量，其形状为 [batch_size,512] ，我有一个常量矩阵，其值仅为0和1，其形状为 [256,512] . 我想为每个批次有效地计算我的向量的产品总和（数据张量的第二维）仅对于1而不是0的条目 . 一个解释的例子：让我们说我有1个大小的批处理：数据张量的值为 [5,4,3,7,8,2] ，我的常量矩阵具有以下值： [0,1,1,0,0,0] [1,0,0,0,0,0] [...
    python tensorflow binary-matrix

热门问题