Is your feature request related to a problem? Please describe.
Need to support the case that the image is a list of tensors and label is only 1 tensor during metrics computation.