下载
中文
注册

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

函数功能

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

函数原型

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

约束说明

索引需要在有效范围中。

参数说明

参数名

参数类型

输入/输出

说明

index

int

输入

获得对应的输出。这里的int范围对应C++中的int。