Cast
函数功能
算子converter中的构图元素，表示一个Cast节点，即图中Tensor的类型转换方法。
函数原型
def Cast(x: Tensor, *, dst_type: int, dependencies=[], node_name=None) -> Tensor
参数说明
参数 |
输入/输出 |
说明 |
是否必选 |
|---|---|---|---|
x |
输入 |
待转换的Tensor。 |
是 |
dst_type |
输入 |
Tensor转换后的数据类型，类型取值参见DataType类。 |
是 |
dependencies |
输入 |
用于指定节点的控制边，标识该节点在图中的拓扑依赖关系，传入时列表中为Tensor类型。 |
否 |
node_name |
输入 |
节点名，缺省为None，例如'const_1'，同一张图中节点名不允许重复。 |
否 |
* |
输入 |
预留参数项，用于后续功能拓展。 |
否 |
返回值说明
正常情况下，返回新类型的Tensor，否则失败报错。
约束说明
无
调用示例
1 2 3 | import torch_npu, torchair from torchair.ge import DataType Cast(1., dst_type=DataType.DT_INT64) |
父主题： GE相关类和接口