昇腾社区首页
中文
注册
开发者
下载

InputTensorInfo

输入Tensor信息,头文件位于CANN软件安装后文件存储路径下的include/ge/ge_api_types.h。

1
2
3
4
5
6
struct InputTensorInfo {
  uint32_t data_type;               // data type
  std::vector<int64_t> dims;        // shape description
  void *data;                       // tensor data
  int64_t length;                   // tensor length
};