Tensorflow根据元素和元素的索引创建matrx

时间:2018-05-08 12:42:01

标签: python matrix tensorflow tensor

比如说,对于N-by-N矩阵,只有m个元素是非零的。如果我们已经有m个非零元素及其所有索引(i,j),那么如何在Tensorflow中构建矩阵?

1 个答案:

答案 0 :(得分:1)

使用tf.SparseTensor,如果需要,还可以使用tf.sparse_tensor_to_dense。例如:

let hasSwimmerPrefixedWithA = events.filter{ $0.swimmers?.contains(where: {$0.swimmer.hasPrefix("A")}) ?? false }