简介
Shape结构体用于描述一个tensor的shape,包含两个信息:
1 2 | size_t dim_num_; int64_t dims_[kMaxDimNum]; |
其中,dim_num_表示shape的维数,dims_数组表示tensor具体的shape。
父主题: Shape
Shape结构体用于描述一个tensor的shape,包含两个信息:
1 2 | size_t dim_num_; int64_t dims_[kMaxDimNum]; |
其中,dim_num_表示shape的维数,dims_数组表示tensor具体的shape。