3
例如, 我有一个二维张量。添加带偏移量的张量
array = np.array([1,2,3],[1,2,3],[1,2,3])
tensor = tf.constant(array)
我想计算张量中每列的总和,但要求每一行与其他行相互抵消。
[0,1,2,3,0,0]
[0,0,1,2,3,0]
[0,0,0,1,2,3]
是否有可能有效地执行这种计算?我试图用SparseTensor
和tf.sparse_reduce_sum
计算稀疏表示的总和,但发现它对我的数据大小而言计算效率低下。
hm ....如果'tf.diag_part'支持像'numpy.diagonal'这样的'offset'参数, –