Skip to content

added squeeze to get rid of excess dimension and fixed shape #5877

added squeeze to get rid of excess dimension and fixed shape

added squeeze to get rid of excess dimension and fixed shape #5877