def __init__(self: ascendie.PermutationConf, arg0: int, arg1: buffer) -> None
函数功能
用于存储排列配置信息。
函数原型
__init__(self: ascendie.PermutationConf, arg0: int, arg1: buffer) -> None
参数说明
参数名 |
参数类型 |
输入/输出 |
说明 |
|---|---|---|---|
rank |
int |
输入 |
表达操作张量的rank。这里的int对应C++中的int64_t。 |
order |
buffer |
输入 |
transpose后各轴对应的排布。例如：PermutationConf { 3, {1, 2, 0}}，可以将一个CHW的tensor调整为HWC。这里的每一维int对应C++中的int64_t。 |