更新时间:2021-03-18 GMT+08:00
分享

计算接口列表

TBE DSL提供的计算接口主要涵盖向量运算,包括Element-wise类操作接口、Reduction 操作接口、Broadcast操作接口、Index操作接口、Concat操作接口、卷积接口、4D/5D互转接口、矩阵计算接口等。

您可以在ATC的安装目录下的“/python/site-packages”目录下查看接口的定义文件。

详细的接口介绍可参见TBE DSL API

分享:

    相关文档

    相关产品

close