open3d.ml.tf.models.global_average

open3d.ml.tf.models.global_average(x, batch_lengths)

Block performing a global average over batch pooling :param x: [N, D] input features :param batch_lengths: [B] list of batch lengths :return: [B, D] averaged features