8a801031c5a5f34f5d83319aa99a9e0889bb56a1,gpflow/transforms.py,LowerTriangular,backward_tensor,#LowerTriangular#Any#,392

Before Change

    def backward_tensor(self, y):
        N = tf.cast(tf.sqrt(tf.size(y) / self.num_matrices), tf.int32)
        reshaped = tf.reshape(y, (N, N, self.num_matrices))
        size = len(reshaped)
        triangular = reshaped[np.tril_indices(size, 0)].T
        return triangular

    def log_jacobian_tensor(self, x):

After Change

        N = tf.cast(tf.sqrt(tf.size(y) / self.num_matrices), tf.int32)
        reshaped = tf.reshape(y, shape=(N, N, self.num_matrices))
        print(reshaped, reshaped.shape.as_list())
        N_not_tensor = reshaped.shape.as_list[0]
        indices = np.dstack(np.tril_indices(N_not_tensor))[0]
        triangular = tf.reshape(tf.gather_nd(reshaped, indices), shape=[-1])
        return triangular

    def log_jacobian_tensor(self, x):
        return tf.zeros((1,), settings.float_type)

Instances