MSTensor

import com.mindspore.lite.MSTensor;

MSTensor定义了MindSpore Lite中的张量。

公有成员函数

function
int[] getShape()
int getDataType()
byte[] getByteData()
float[] getFloatData()
int[] getIntData()
long[] getLongData()
void setData(byte[] data)
void setData(ByteBuffer data)
long size()
int elementsNum()
void free()

getShape

public int[] getShape()

获取MindSpore Lite MSTensor的形状。

• 返回值

  一个包含MindSpore Lite MSTensor形状数值的整型数组。

getDataType

public int getDataType()

DataType在com.mindspore.lite.DataType中定义。

• 返回值

  MindSpore Lite MSTensor类的MindSpore Lite DataType。

getByteData

public byte[] getByteData()

获得MSTensor的输出数据，数据类型为byte类型。

• 返回值

  包含所有MSTensor输出数据的byte类型数组。

getFloatData

public float[] getFloatData()

获得MSTensor的输出数据，数据类型为float类型。

• 返回值

  包含所有MSTensor输出数据的float类型数组。

getIntData

public int[] getIntData()

获得MSTensor的输出数据，数据类型为int类型。

• 返回值

  包含所有MSTensor输出数据的int类型数组。

getLongData

public long[] getLongData()

获得MSTensor的输出数据，数据类型为long类型。

• 返回值

  包含所有MSTensor输出数据的long类型数组。

setData

public void setData(byte[] data)

设定MSTensor的输入数据。

• 参数

  • data: byte[]类型的输入数据。

public void setData(ByteBuffer data)

设定MSTensor的输入数据。

• 参数

  • data: ByteBuffer类型的输入数据。

size

public long size()

获取MSTensor中的数据的字节数大小。

• 返回值

  MSTensor中的数据的字节数大小。

elementsNum

public int elementsNum()

获取MSTensor中的元素个数。

• 返回值

  MSTensor中的元素个数。

free

public void free()

释放MSTensor运行过程中动态分配的内存。