def add_input(self: Network, name: str, type: DataType, dim: Dims) -> Tensor

函数功能

向网络添加输入张量。

函数原型

add_input(self: ascendie.pyascendie.Network, arg0: str, arg1: ascendie.pyascendie.DataType, arg2: ascendie.pyascendie.Dims) -> ascendie.pyascendie.Tensor

约束说明

参数说明

参数名

参数类型

输入/输出

说明

name

String

输入

张量的名称。

type

DataType

输入

张量中保存的数据类型。

dim

Dims

输入

张量的维度。