Error: There was a problem grading your submission. Details: sub got incompatible shapes for broadcasting: (5,), (50,)

closing this thread as his issue is addressed in his created post