MSTensor

import com.mindspore.MSTensor;

MSTensor定义了MindSpore中的张量。

公有成员函数

function	云侧推理是否支持	端侧推理是否支持
MSTensor createTensor(String tensorName, int dataType, int[] tensorShape, ByteBuffer buffer)	√	√
MSTensor createTensor(String tensorName, Object obj)	√	√
int[] getShape()	√	√
int getDataType()	√	√
Object getData()	√	√
byte[] getByteData()	√	√
float[] getFloatData()	√	√
int[] getIntData()	√	√
long[] getLongData()	√	√
boolean setShape(int[] tensorShape)	√	√
boolean setData(byte[] data)	√	√
boolean setData(float[] data)	√	√
boolean setData(int[] data)	√	√
boolean setData(long[] data)	√	√
boolean setData(ByteBuffer data)	√	√
long size()	√	√
int elementsNum()	√	√
void free()	√	√
String tensorName()	√	√
long getMSTensorPtr()	√	√
DataType	√	√

createTensor

public static MSTensor createTensor(String tensorName, int dataType, int[] tensorShape, ByteBuffer buffer)

生成MindSpore MSTensor。

参数
- tensorName: 张量名称。
- dataType: 张量数据类型。
- tensorShape: 张量形状。
- buffer: 张量数据。
返回值

MindSpore MSTensor。

public static MSTensor createTensor(String tensorName, Object obj)

生成MindSpore MSTensor。

参数
- tensorName: 张量名称。
- obj: java的Array对象或者一个标量值，支持的数据类型：float、double、int、long、boolean。
返回值

MindSpore MSTensor。

getShape

public int[] getShape()

获取MindSpore MSTensor的形状。

返回值

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

getDataType

public int getDataType()

DataType在com.mindspore.DataType中定义。

返回值

MindSpore MSTensor类的MindSpore DataType。

getData

public Object getData()

获取MSTensor的输出数据，数据类型与设置的类型数据相同。

返回值

包含所有MSTensor输出数据的字节数组。

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类型数组。

setShape

public boolean setShape(int[] tensorShape)

设定MSTensor的形状。

参数
- tensorShape： int[]类型的输入数据。
返回值

设置形状是否成功。

setData

public boolean setData(byte[] data)

设定MSTensor的输入数据。

参数
- data: byte[]类型的输入数据。
返回值

设置数据是否成功。

public boolean setData(float[] data)

设定MSTensor的输入数据。

参数
- data: float[]类型的输入数据。
返回值

设置数据是否成功。

public boolean setData(int[] data)

设定MSTensor的输入数据。

参数
- data: int[]类型的输入数据。
返回值

设置数据是否成功。

public boolean setData(long[] data)

设定MSTensor的输入数据。

参数
- data: long[]类型的输入数据。
返回值

设置数据是否成功。

public boolean setData(ByteBuffer data)

设定MSTensor的输入数据。

参数
- data: ByteBuffer类型的输入数据。
返回值

设置数据是否成功。

size

public long size()

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

返回值

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

elementsNum

public int elementsNum()

获取MSTensor中的元素个数。

返回值

MSTensor中的元素个数。

free

public void free()

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

tensorName

public String tensorName()

返回tensor的名称。

返回值

tensor的名称。

getMSTensorPtr

public long getMSTensorPtr()

返回MStensor对象的指针。

返回值

MStensor对象的指针。

DataType

import com.mindspore.config.DataType;

DataType定义了MindSpore中的张量的数据类型。

公有成员变量

public static final int kNumberTypeBool = 30;
public static final int kNumberTypeInt = 31;
public static final int kNumberTypeInt8 = 32;
public static final int kNumberTypeInt16 = 33;
public static final int kNumberTypeInt32 = 34;
public static final int kNumberTypeInt64 = 35;
public static final int kNumberTypeUInt = 36;
public static final int kNumberTypeUInt8 = 37;
public static final int kNumberTypeUInt16 = 38;
public static final int kNumberTypeUint32 = 39;
public static final int kNumberTypeUInt64 = 40;
public static final int kNumberTypeFloat = 41;
public static final int kNumberTypeFloat16 = 42;
public static final int kNumberTypeFloat32 = 43;
public static final int kNumberTypeFloat64 = 44;