首页 > 解决方案 > 3D tf.Tensor 中的零之间的总和

问题描述

我有一个形状为 (10, 100, 3) 的 tf.Tensor 数据集,如下所示。

[[[ 0.    0.    0.  ],
  [-0.07 -0.07  0.02],
  [-0.04 -0.05 -0.00],
  [ 0.    0.    0.  ],
  ...
  [ 0.    0.    0.  ],
  [ 0.01  0.02  0.12],
  [-0.04 -0.00  0.01],
  [-0.00 -0.06  0.04],
  [ 0.    0.    0.  ],
  [ 0.    0.    0.  ]], #(1, 100, 3)

 [[continue .  ...  ],
  [  ...  ...  ...  ],
  [  ...  ...  ...  ]], #(1, 100, 3)

 [[continue 8 more  ],
  [  ...  ...  ...  ],
  [  ...  ...  ...  ]]] #(8, 100, 3)

毕竟,数据集的形状就像 (10, ##, 3) 不规则张量。如何在不使用循环的情况下对零向量之间的所有非零向量求和?

标签: python-3.xtensorflowkeras

解决方案


推荐阅读