昇腾社区首页
中文
注册

def get_input(self: ascendie.BaseLayer, arg0: int) -> ascendie.Tensor

函数功能

获取指定索引位置的输入张量。

函数原型

get_input(self: ascendie.BaseLayer, arg0: int) -> ascendie.Tensor

约束说明

索引需要在有效范围中。

参数说明

参数名

参数类型

输入/输出

说明

index

int

输入

对应layer输入索引位置。这里的int范围对应C++中的int。