数据存储方法和装置制造方法及图纸

本发明专利技术涉及计算机技术领域，尤其涉及一种数据存储方法和装置。先获取待转换张量，待转换张量包含通道维度、宽度维度、高度维度和数量维度，然后根据待转换张量配置第一张量，第一张量在通道维度和数量维度方向上的张量数据的个数均为第一数值的倍数，最后将第一张量以第一存储格式存储至存储设备，以第一存储格式存储的第一张量读取张量数据时，先在通道维度方向上进行读取，且每次读取张量数据的个数为第一数值。配置了第一存储格式，并提供了其他存储格式到第一存储格式具体步骤，提高了数据的读取速度。据的读取速度。据的读取速度。

【技术实现步骤摘要】
数据存储方法和装置


[0001]本专利技术涉及计算机
，尤其涉及一种数据存储方法和装置。

技术介绍

[0002]常见的卷积神经网络中，张量由数量维度N、高度维度H、宽度维度W和通道维度C四个维度构成，根据不同的数据存储格式，张量数据在存储器内按不同的维度方向存储为一维数据。在卷积计算过程中，为了提高计算效率，可以并行进行8个卷积乘计算。计算时取得张量数据的基本方法是，每次在通道维度C上取8个数据。但是，对于一些不以通道维度(C方向)开始存储的数据存储格式，如NCHW、HWCN等，如果按照通道维度(C方向)读取数据，由于数据排布在通道维度(C方向)并不连续，会导致操作次数多且查找复杂。

技术实现思路

[0003]本专利技术实施例提供了一种数据存储方法和装置，将原来的待转换张量扩充为第一张量，再将第一张量以第一存储格式存储至存储设备，使得存储的第一张量能够先在通道维度方向上读取张量数据，并指定每次读取的个数，提高了读取张量数据的速度。
[0004]第一方面，本专利技术实施例提供了一种数据存储方法，包括：
[0005]获取待转换张量，所述待转换张量包含通道维度、宽度维度、高度维度和数量维度；
[0006]根据所述待转换张量配置第一张量，所述第一张量在所述通道维度和所述数量维度方向上的张量数据的个数均为第一数值的倍数；
[0007]将所述第一张量以第一存储格式存储至存储设备，以所述第一存储格式存储的所述第一张量读取张量数据时，先在所述通道维度方向上进行读取，且每次读取所述张量数据的个数为所述第一数值。
[0008]一种实现方式中，所述根据所述待转换张量配置第一张量，包括：
[0009]以0作为张量数据对所述待转换张量的所述数量维度和所述通道维度进行扩充，配置所述第一张量。
[0010]一种实现方式中，所述根据所述待转换张量配置第一张量，包括：
[0011]获取所述待转换张量在通道维度方向上的张量个数C1和在数量维度方向上的张量个数N1；
[0012]根据公式C2＝C1/M确定C1与M的商，所述M为所述第一数值；
[0013]根据公式C3＝(C2+1)
×
M，确定所述第一张量在通道维度方向上的张量数据个数；
[0014]根据公式N2＝N1/M确定N1与M的商；
[0015]根据公式N3＝(N2+1)
×
M，确定所述第一张量在数量维度方向上的张量数据个数。
[0016]一种实现方式中，将所述第一张量以第一存储格式存储至所述存储设备，包括：
[0017]获取第二存储格式，所述待转换张量以所述第二存储格式在所述存储设备中存储；
[0018]根据所述第一存储格式和所述第二存储格式，确定所述第一张量中各张量数据的数据序号；
[0019]将所述第一张量中的张量数据按所述数据序号的顺序存储至所述存储设备。
[0020]一种实现方式中，所述根据所述第一存储格式和所述第二存储格式，确定所述第一张量中各张量数据的数据序号，包括：
[0021]将所述数据序号的初始值设置为0；
[0022]按指定存储规则遍历所述第一张量，将所诉数据序号与对应的张量数据关联，每遍历一个张量数据，所述数据序号的值加一。
[0023]一种实现方式中，所述第一存储格式包括六维排列N
x
HWC
x
NC格式；
[0024]所述按指定存储规则遍历所述第一张量，将所诉数据序号与对应的张量数据关联，包括：
[0025]步骤1，在通道维度方向上遍历x次，每次遍历一个张量数据；
[0026]步骤2，在数量维度方向上遍历x次，每次遍历步骤1；
[0027]步骤3，返回数量维度方向上的初始位置并顺序执行步骤1和步骤2；
[0028]步骤4，循环执行步骤3，直至通道维度方向上的张量数据遍历完毕；
[0029]步骤5，在宽度维度方向上遍历，每次遍历顺序执行步骤1至步骤4，直至宽度维度方向上的张量数据遍历完毕；
[0030]步骤6，在高度维度方向上遍历，每次遍历顺序执行步骤1至步骤5，直至高度维度方向上的张量数据遍历完毕；
[0031]步骤7，在数量维度方向上遍历，每次遍历顺序执行步骤1至步骤6，直至所有张量数据遍历完毕；
[0032]其中，所述x为所述第一数值。
[0033]一种实现方式中，所述第一存储格式包括四维排列N
y
HWC
y
格式；
[0034]所述按指定存储规则遍历所述第一张量，将所诉数据序号与对应的张量数据关联，包括：
[0035]步骤1，在通道维度方向上遍历y次，每次遍历一个张量数据；
[0036]步骤2，在数量维度方向上遍历，每次遍历执行步骤1，直至数量维度方向上的张量数据遍历完毕；
[0037]步骤3，返回数量维度方向上的初始位置，顺序执行步骤1和步骤2；
[0038]步骤4，循环执行步骤3，直至通道维度方向上的张量数据遍历完毕；
[0039]步骤5，在宽度维度方向上遍历，每次遍历顺序执行步骤1至步骤4，直至宽度维度方向上的张量数据遍历完毕；
[0040]步骤6，在高度维度方向上遍历，每次遍历顺序执行步骤1至步骤5，直至所有张量数据遍历完毕；
[0041]其中，y为所述第一数值。
[0042]第二方面，本专利技术实施例提供了一种数据存储装置，包括：
[0043]获取模块，用于获取待转换张量，所述待转换张量包含通道维度、宽度维度、高度维度和数量维度；
[0044]配置模块，用于根据所述待转换张量配置第一张量，所述第一张量在所述通道维
度和所述数量维度方向上的张量数据的个数均为第一数值的倍数；
[0045]存储模块，用于将所述第一张量以第一存储格式存储至存储设备，以所述第一存储格式存储的所述第一张量读取张量数据时，先在所述通道维度方向上进行读取，且每次读取所述张量数据的个数为所述第一数值。
[0046]第三方面，本专利技术实施例提供了一种电子芯片，包括：
[0047]至少一个处理器；以及
[0048]与所述处理器通信连接的至少一个存储器，其中：
[0049]所述存储器存储有可被所述处理器执行的程序指令，所述处理器调用所述程序指令能够执行第一方面提供的方法。
[0050]第四方面，本专利技术实施例提供了一种计算机可读存储介质，所述计算机可读存储介质包括存储的程序，其中，在所述程序运行时控制所述计算机可读存储介质所在设备执行第一方面提供的方法。
[0051]本专利技术实施例中，先获取待转换张量，待转换张量包含通道维度、宽度维度、高度维度和数量维度，然后根据待转换张量配置第一张量，第一张量在通道维度和数量维度方向上的张量数据的个数均为第一数值的倍数，最后将第一张量以第一存储格式存储至存储设备，以第一存储格式存储的第一张量先在通道维度方本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种数据存储方法，其特征在于，包括：获取待转换张量，所述待转换张量包含通道维度、宽度维度、高度维度和数量维度；根据所述待转换张量配置第一张量，所述第一张量在所述通道维度和所述数量维度方向上的张量数据的个数均为第一数值的倍数；将所述第一张量以第一存储格式存储至存储设备，以所述第一存储格式存储的所述第一张量读取张量数据时，先在所述通道维度方向上进行读取，且每次读取所述张量数据的个数为所述第一数值。2.根据权利要求1所述的方法，其特征在于，所述根据所述待转换张量配置第一张量，包括：以0作为张量数据对所述待转换张量的所述数量维度和所述通道维度进行扩充，配置所述第一张量。3.根据权利要求2所述的方法，其特征在于，所述根据所述待转换张量配置第一张量，包括：获取所述待转换张量在通道维度方向上的张量个数C1和在数量维度方向上的张量个数N1；根据公式C2＝C1/M确定C1与M的商，所述M为所述第一数值；根据公式C3＝(C2+1)
×
M，确定所述第一张量在通道维度方向上的张量数据个数；根据公式N2＝N1/M确定N1与M的商；根据公式N3＝(N2+1)
×
M，确定所述第一张量在数量维度方向上的张量数据个数。4.根据权利要求1所述的方法，其特征在于，将所述第一张量以第一存储格式存储至所述存储设备，包括：获取第二存储格式，所述待转换张量以所述第二存储格式在所述存储设备中存储；根据所述第一存储格式和所述第二存储格式，确定所述第一张量中各张量数据的数据序号；将所述第一张量中的张量数据按所述数据序号的顺序存储至所述存储设备。5.根据权利要求4所述的方法，其特征在于，所述根据所述第一存储格式和所述第二存储格式，确定所述第一张量中各张量数据的数据序号，包括：将所述数据序号的初始值设置为0；按指定存储规则遍历所述第一张量，将所诉数据序号与对应的张量数据关联，每遍历一个张量数据，所述数据序号的值加一。6.根据权利要求5所述的方法，其特征在于，所述第一存储格式包括六维排列N
x
HWC
x
NC格式；所述按指定存储规则遍历所述第一张量，将所诉数据序号与对应的张量数据关联，包括：步骤1，在通道维度方向上遍历x次，每次遍历一个张量数据；步骤2，在数量维度方向上遍历x次，每次遍历执行步骤1；步骤3，...

【专利技术属性】
技术研发人员：孙星，宋玉超，
申请(专利权)人：展讯通信天津有限公司，
类型：发明
国别省市：