sponge.function.check_broadcast

查看源文件
sponge.function.check_broadcast(shape0: tuple, shape1: tuple)[源代码]

检查两个shapes是否符合播出规则。

参数:
  • shape0 (tuple) - 第一个shape

  • shape1 (tuple) - 第二个shape

返回:

tuple。shape。播出之后的shape。

支持平台:

Ascend GPU CPU