Tensorflow：给定 True 索引创建布尔矩阵

Question

假设我有以下两个向量，一个用于开始索引，另一个用于结束索引start_index = [1, 1, 2]end_index = [3, 4, 3]在下面我有最终布尔矩阵的形状shape = [3, 6]我想生成以下布尔矩阵bool_mat = [[False, True,  True, True, False, False]            [False, True,  True, True, True,  False]            [False, False, True, True, False, False]]对于每一行，True 从 start_index 中的索引开始，以 end_index 中的索引结束，其他地方为 False，bool_mat[i, start_index[i]:end_index[i]+1] = True如何在 TensorFlow 中做到这一点？谢谢！