下载
中文
注册

struct PermutationConf

说明

用来定义Tensor的transpose配置。

定义

struct PermutationConf {
    static constexpr uint64_t MAX_DIM = 8;
    int64_t rank;
    int64_t order[MAX_DIM];
};

成员

成员名称

描述

rank

表达操作张量的rank。

order

transpose后各轴对应的排布。例如:PermutationConf { 3, {1, 2, 0}},可以将一个CHW的tensor调整为HWC。