Week 3 programming assignment error

Why is my code not passing the testcase? Have done everything as asked.

import tensorflow as tf

def compute_total_loss(logits, labels):
“”"
Computes the total loss

Arguments:
logits -- output of forward propagation (output of the last LINEAR unit), of shape (6, num_examples)
labels -- "true" labels vector, same shape as Z3

Returns:
total_loss - Tensor of the total loss value
"""

# Set from_logits to True for categorical cross-entropy
total_loss = tf.reduce_sum(tf.keras.losses.categorical_crossentropy(labels, logits, from_logits=True))

return total_loss

OUTPUT
tf.Tensor(0.17102128, shape=(), dtype=float32)
AssertionError: Test does not match. Did you get the reduce sum of your loss functions?

If I recall correctly, you need to also transpose the labels and logits variables.