Open Bihaqo opened 5 years ago
Something like
def sparse(idx, n): cores = [] eye = tf.eye(n) for k in range(len(idx)): cores.append(tf.reshape(eye[idx[k]], (1, n, 1))) return t3f.TensorTrain(cores)
Something like