BroadcastTo【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√Atlas 200I/500 A2 推理产品√Atlas 推理系列产品√Atlas 训练系列产品√功能说明算子功能将输入tensor广播到指定的shape。如输入tensor的shape为(1, 4)指定的shape为(2, 4)则输出是shape为(2, 4)的tensor。参数说明参数名输入/输出/属性描述数据类型数据格式x输入张量需要被广播的张量。BF16、FLOAT16、FLOAT、 DOUBLE、INT64、INT32、UINT8、UINT16、UINT32、UINT64、INT8、INT16、COMPLEX32、COMPLEX64、COMPLEX128、QINT8、QUINT8、QINT16、QUINT16、QINT32、BOOL、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN、STRINGNDshape输入张量表示 x 广播后的shape大小。INT64、INT32-y输出维度最大不超过8维shape由shape输入决定dtype需要与self一致。同 xND约束说明无调用说明调用方式样例代码说明图模式调用test_geir_broadcast_to通过算子IR构图方式调用broadcast_to算子。【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考