Attend the Women in ML Symposium on December 7 Register now

Stay organized with collections Save and categorize content based on your preferences.

Zero out IDs of classes not in allowed_class_ids.

batch_class_ids A [batch_size, num_instances] int tensor of input class IDs.
allowed_class_ids A python list of class IDs which we want to allow.

filtered_class_ids A [batch_size, num_instances] int tensor with any class ID not in allowed_class_ids set to 0.