昇腾社区首页
中文
注册

def set_input(self: ascendie.BaseLayer, arg0: int, arg1: ascendie.Tensor)

函数功能

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

函数原型

set_input(self: ascendie.BaseLayer, arg0: int, arg1: ascendie.Tensor)

约束说明

索引需要在有效范围中。

参数说明

参数名

参数类型

输入/输出

说明

index

int

输入

所设置输入张量对应的索引。这里的int范围对应C++中的uint32_t。

tensor

Tensor

输入

需要设置为输入的张量。