MSTensor
import com.mindspore.MSTensor;
MSTensor定义了MindSpore中的张量。
公有成员函数
function |
---|
MSTensor createTensor(String tensorName, int dataType, int[] tensorShape, ByteBuffer buffer) |
createTensor
public static MSTensor createTensor(String tensorName, int dataType, int[] tensorShape, ByteBuffer buffer)
生成MindSpore MSTensor。
参数
tensorName
: 张量名称。dataType
: 张量数据类型。tensorShape
: 张量形状。buffer
: 张量数据。
返回值
MindSpore MSTensor。
getShape
public int[] getShape()
获取MindSpore MSTensor的形状。
返回值
一个包含MindSpore MSTensor形状数值的整型数组。
getDataType
public int getDataType()
DataType在com.mindspore.DataType中定义。
返回值
MindSpore MSTensor类的MindSpore 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(float[] data)
设定MSTensor的输入数据。
参数
data
: float[]类型的输入数据。
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运行过程中动态分配的内存。
tensorName
public String tensorName()
返回tensor的名称。
返回值
tensor的名称。
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;