sponge.function.check_broadcast

View Source On Gitee
sponge.function.check_broadcast(shape0: tuple, shape1: tuple)[source]

Check whether the two shapes match the rule of broadcast.

Parameters

  • shape0 (tuple) – First shape

  • shape1 (tuple) – Second shape

Returns

shape (tuple), Shape after broadcast

Supported Platforms:

Ascend GPU CPU