使用 Functional 操作与计算#

备注

  • 除了 Tensor 内置的方法,我们还可以调用 Functional 包中实现的有关操作与计算接口。

  • 想要快速浏览或检索所有相关 API 和例程,请参考 megengine.functional 页面。

注意 Tensor 数据类型

Tensor 之间进行计算时需要考虑到对输入数据类型的支持情况,通常默认输入 Tesnor 为 float32 类型, 将不支持的数据类型 Tensor 作为输入可能将无法进行计算,请参考 Tensor 数据类型 中的介绍。

术语概念区分

大部分时候,我们会将对 Tensor 的操作和运算区分为两个概念:

  • 对单个 Tensor 进行操作(Manipulation),通常其属性(如形状、设备、类型等)发生了变化;

  • Tensor 之间可以进行运算(Operation),通常基于 Tensor 内元素的数值进行了计算。

而有些时候它们概念一致,彼此之间可以互相指代(取决于上下文)。 但可以确定的一点是,不论如何称呼这些行为,它们都可以从 functional 包中能找到, 也可用 “算子(Operators)” 来称呼相关行为。

区别于 API 参考中的分类方式,本文档中将 Tensor 的常见操作/运算分为以下类型进行介绍: