Package onnx

Class OnnxMl.TensorProto

  • All Implemented Interfaces:
    Serializable, OnnxMl.TensorProtoOrBuilder, org.nd4j.shade.protobuf.Message, org.nd4j.shade.protobuf.MessageLite, org.nd4j.shade.protobuf.MessageLiteOrBuilder, org.nd4j.shade.protobuf.MessageOrBuilder
    Enclosing class:
    OnnxMl
    public static final class OnnxMl.TensorProto
extends org.nd4j.shade.protobuf.GeneratedMessageV3
implements OnnxMl.TensorProtoOrBuilder
     Tensors
 A serialized tensor value.
    Protobuf type onnx.TensorProto
    See Also:
    Serialized Form

    • Method Detail

      • newInstance

        protected Object newInstance​(org.nd4j.shade.protobuf.GeneratedMessageV3.UnusedPrivateParameter unused)
        Overrides:
        newInstance in class org.nd4j.shade.protobuf.GeneratedMessageV3

      • getUnknownFields

        public final org.nd4j.shade.protobuf.UnknownFieldSet getUnknownFields()
        Specified by:
        getUnknownFields in interface org.nd4j.shade.protobuf.MessageOrBuilder
        Overrides:
        getUnknownFields in class org.nd4j.shade.protobuf.GeneratedMessageV3

      • getDescriptor

        public static final org.nd4j.shade.protobuf.Descriptors.Descriptor getDescriptor()

      • internalGetFieldAccessorTable

        protected org.nd4j.shade.protobuf.GeneratedMessageV3.FieldAccessorTable internalGetFieldAccessorTable()
        Specified by:
        internalGetFieldAccessorTable in class org.nd4j.shade.protobuf.GeneratedMessageV3

      • getDims

        public long getDims​(int index)
         The shape of the tensor.
        repeated int64 dims = 1;
        Specified by:
        getDims in interface OnnxMl.TensorProtoOrBuilder
        Parameters:
        index - The index of the element to return.
        Returns:
        The dims at the given index.

      • getDataType

        public int getDataType()
         The data type of the tensor.
 This field MUST have a valid TensorProto.DataType value
        int32 data_type = 2;
        Specified by:
        getDataType in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        The dataType.

      • hasSegment

        public boolean hasSegment()
        .onnx.TensorProto.Segment segment = 3;
        Specified by:
        hasSegment in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        Whether the segment field is set.

      • getFloatDataList

        public List<Float> getFloatDataList()
         For float and complex64 values
 Complex64 tensors are encoded as a single array of floats,
 with the real components appearing in odd numbered positions,
 and the corresponding imaginary component appearing in the
 subsequent even numbered position. (e.g., [1.0 + 2.0i, 3.0 + 4.0i]
 is encoded as [1.0, 2.0 ,3.0 ,4.0]
 When this field is present, the data_type field MUST be FLOAT or COMPLEX64.
        repeated float float_data = 4 [packed = true];
        Specified by:
        getFloatDataList in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        A list containing the floatData.

      • getFloatDataCount

        public int getFloatDataCount()
         For float and complex64 values
 Complex64 tensors are encoded as a single array of floats,
 with the real components appearing in odd numbered positions,
 and the corresponding imaginary component appearing in the
 subsequent even numbered position. (e.g., [1.0 + 2.0i, 3.0 + 4.0i]
 is encoded as [1.0, 2.0 ,3.0 ,4.0]
 When this field is present, the data_type field MUST be FLOAT or COMPLEX64.
        repeated float float_data = 4 [packed = true];
        Specified by:
        getFloatDataCount in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        The count of floatData.

      • getFloatData

        public float getFloatData​(int index)
         For float and complex64 values
 Complex64 tensors are encoded as a single array of floats,
 with the real components appearing in odd numbered positions,
 and the corresponding imaginary component appearing in the
 subsequent even numbered position. (e.g., [1.0 + 2.0i, 3.0 + 4.0i]
 is encoded as [1.0, 2.0 ,3.0 ,4.0]
 When this field is present, the data_type field MUST be FLOAT or COMPLEX64.
        repeated float float_data = 4 [packed = true];
        Specified by:
        getFloatData in interface OnnxMl.TensorProtoOrBuilder
        Parameters:
        index - The index of the element to return.
        Returns:
        The floatData at the given index.

      • getInt32DataList

        public List<Integer> getInt32DataList()
         For int32, uint8, int8, uint16, int16, bool, and float16 values
 float16 values must be bit-wise converted to an uint16_t prior
 to writing to the buffer.
 When this field is present, the data_type field MUST be
 INT32, INT16, INT8, UINT16, UINT8, BOOL, or FLOAT16
        repeated int32 int32_data = 5 [packed = true];
        Specified by:
        getInt32DataList in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        A list containing the int32Data.

      • getInt32DataCount

        public int getInt32DataCount()
         For int32, uint8, int8, uint16, int16, bool, and float16 values
 float16 values must be bit-wise converted to an uint16_t prior
 to writing to the buffer.
 When this field is present, the data_type field MUST be
 INT32, INT16, INT8, UINT16, UINT8, BOOL, or FLOAT16
        repeated int32 int32_data = 5 [packed = true];
        Specified by:
        getInt32DataCount in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        The count of int32Data.

      • getInt32Data

        public int getInt32Data​(int index)
         For int32, uint8, int8, uint16, int16, bool, and float16 values
 float16 values must be bit-wise converted to an uint16_t prior
 to writing to the buffer.
 When this field is present, the data_type field MUST be
 INT32, INT16, INT8, UINT16, UINT8, BOOL, or FLOAT16
        repeated int32 int32_data = 5 [packed = true];
        Specified by:
        getInt32Data in interface OnnxMl.TensorProtoOrBuilder
        Parameters:
        index - The index of the element to return.
        Returns:
        The int32Data at the given index.

      • getStringDataList

        public List<org.nd4j.shade.protobuf.ByteString> getStringDataList()
         For strings.
 Each element of string_data is a UTF-8 encoded Unicode
 string. No trailing null, no leading BOM. The protobuf "string"
 scalar type is not used to match ML community conventions.
 When this field is present, the data_type field MUST be STRING
        repeated bytes string_data = 6;
        Specified by:
        getStringDataList in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        A list containing the stringData.

      • getStringDataCount

        public int getStringDataCount()
         For strings.
 Each element of string_data is a UTF-8 encoded Unicode
 string. No trailing null, no leading BOM. The protobuf "string"
 scalar type is not used to match ML community conventions.
 When this field is present, the data_type field MUST be STRING
        repeated bytes string_data = 6;
        Specified by:
        getStringDataCount in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        The count of stringData.

      • getStringData

        public org.nd4j.shade.protobuf.ByteString getStringData​(int index)
         For strings.
 Each element of string_data is a UTF-8 encoded Unicode
 string. No trailing null, no leading BOM. The protobuf "string"
 scalar type is not used to match ML community conventions.
 When this field is present, the data_type field MUST be STRING
        repeated bytes string_data = 6;
        Specified by:
        getStringData in interface OnnxMl.TensorProtoOrBuilder
        Parameters:
        index - The index of the element to return.
        Returns:
        The stringData at the given index.

      • getInt64DataList

        public List<Long> getInt64DataList()
         For int64.
 When this field is present, the data_type field MUST be INT64
        repeated int64 int64_data = 7 [packed = true];
        Specified by:
        getInt64DataList in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        A list containing the int64Data.

      • getInt64DataCount

        public int getInt64DataCount()
         For int64.
 When this field is present, the data_type field MUST be INT64
        repeated int64 int64_data = 7 [packed = true];
        Specified by:
        getInt64DataCount in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        The count of int64Data.

      • getInt64Data

        public long getInt64Data​(int index)
         For int64.
 When this field is present, the data_type field MUST be INT64
        repeated int64 int64_data = 7 [packed = true];
        Specified by:
        getInt64Data in interface OnnxMl.TensorProtoOrBuilder
        Parameters:
        index - The index of the element to return.
        Returns:
        The int64Data at the given index.

      • getNameBytes

        public org.nd4j.shade.protobuf.ByteString getNameBytes()
         Optionally, a name for the tensor.
        string name = 8;
        Specified by:
        getNameBytes in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        The bytes for name.

      • getDocString

        public String getDocString()
         A human-readable documentation for this tensor. Markdown is allowed.
        string doc_string = 12;
        Specified by:
        getDocString in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        The docString.

      • getDocStringBytes

        public org.nd4j.shade.protobuf.ByteString getDocStringBytes()
         A human-readable documentation for this tensor. Markdown is allowed.
        string doc_string = 12;
        Specified by:
        getDocStringBytes in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        The bytes for docString.

      • getRawData

        public org.nd4j.shade.protobuf.ByteString getRawData()
         Serializations can either use one of the fields above, or use this
 raw bytes field. The only exception is the string case, where one is
 required to store the content in the repeated bytes string_data field.
 When this raw_data field is used to store tensor value, elements MUST
 be stored in as fixed-width, little-endian order.
 Floating-point data types MUST be stored in IEEE 754 format.
 Complex64 elements must be written as two consecutive FLOAT values, real component first.
 Complex128 elements must be written as two consecutive DOUBLE values, real component first.
 Boolean type MUST be written one byte per tensor element (00000001 for true, 00000000 for false).
 Note: the advantage of specific field rather than the raw_data field is
 that in some cases (e.g. int data), protobuf does a better packing via
 variable length storage, and may lead to smaller binary footprint.
 When this field is present, the data_type field MUST NOT be STRING or UNDEFINED
        bytes raw_data = 9;
        Specified by:
        getRawData in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        The rawData.

      • getExternalDataList

        public List<OnnxMl.StringStringEntryProto> getExternalDataList()
         Data can be stored inside the protobuf file using type-specific fields or raw_data.
 Alternatively, raw bytes data can be stored in an external file, using the external_data field.
 external_data stores key-value pairs describing data location. Recognized keys are:
 - "location" (required) - POSIX filesystem path relative to the directory where the ONNX
                           protobuf model was stored
 - "offset" (optional) - position of byte at which stored data begins. Integer stored as string.
                         Offset values SHOULD be multiples 4096 (page size) to enable mmap support.
 - "length" (optional) - number of bytes containing data. Integer stored as string.
 - "checksum" (optional) - SHA1 digest of file specified in under 'location' key.
        repeated .onnx.StringStringEntryProto external_data = 13;
        Specified by:
        getExternalDataList in interface OnnxMl.TensorProtoOrBuilder

      • getExternalDataOrBuilderList

        public List<? extends OnnxMl.StringStringEntryProtoOrBuilder> getExternalDataOrBuilderList()
         Data can be stored inside the protobuf file using type-specific fields or raw_data.
 Alternatively, raw bytes data can be stored in an external file, using the external_data field.
 external_data stores key-value pairs describing data location. Recognized keys are:
 - "location" (required) - POSIX filesystem path relative to the directory where the ONNX
                           protobuf model was stored
 - "offset" (optional) - position of byte at which stored data begins. Integer stored as string.
                         Offset values SHOULD be multiples 4096 (page size) to enable mmap support.
 - "length" (optional) - number of bytes containing data. Integer stored as string.
 - "checksum" (optional) - SHA1 digest of file specified in under 'location' key.
        repeated .onnx.StringStringEntryProto external_data = 13;
        Specified by:
        getExternalDataOrBuilderList in interface OnnxMl.TensorProtoOrBuilder

      • getExternalDataCount

        public int getExternalDataCount()
         Data can be stored inside the protobuf file using type-specific fields or raw_data.
 Alternatively, raw bytes data can be stored in an external file, using the external_data field.
 external_data stores key-value pairs describing data location. Recognized keys are:
 - "location" (required) - POSIX filesystem path relative to the directory where the ONNX
                           protobuf model was stored
 - "offset" (optional) - position of byte at which stored data begins. Integer stored as string.
                         Offset values SHOULD be multiples 4096 (page size) to enable mmap support.
 - "length" (optional) - number of bytes containing data. Integer stored as string.
 - "checksum" (optional) - SHA1 digest of file specified in under 'location' key.
        repeated .onnx.StringStringEntryProto external_data = 13;
        Specified by:
        getExternalDataCount in interface OnnxMl.TensorProtoOrBuilder

      • getExternalData

        public OnnxMl.StringStringEntryProto getExternalData​(int index)
         Data can be stored inside the protobuf file using type-specific fields or raw_data.
 Alternatively, raw bytes data can be stored in an external file, using the external_data field.
 external_data stores key-value pairs describing data location. Recognized keys are:
 - "location" (required) - POSIX filesystem path relative to the directory where the ONNX
                           protobuf model was stored
 - "offset" (optional) - position of byte at which stored data begins. Integer stored as string.
                         Offset values SHOULD be multiples 4096 (page size) to enable mmap support.
 - "length" (optional) - number of bytes containing data. Integer stored as string.
 - "checksum" (optional) - SHA1 digest of file specified in under 'location' key.
        repeated .onnx.StringStringEntryProto external_data = 13;
        Specified by:
        getExternalData in interface OnnxMl.TensorProtoOrBuilder

      • getExternalDataOrBuilder

        public OnnxMl.StringStringEntryProtoOrBuilder getExternalDataOrBuilder​(int index)
         Data can be stored inside the protobuf file using type-specific fields or raw_data.
 Alternatively, raw bytes data can be stored in an external file, using the external_data field.
 external_data stores key-value pairs describing data location. Recognized keys are:
 - "location" (required) - POSIX filesystem path relative to the directory where the ONNX
                           protobuf model was stored
 - "offset" (optional) - position of byte at which stored data begins. Integer stored as string.
                         Offset values SHOULD be multiples 4096 (page size) to enable mmap support.
 - "length" (optional) - number of bytes containing data. Integer stored as string.
 - "checksum" (optional) - SHA1 digest of file specified in under 'location' key.
        repeated .onnx.StringStringEntryProto external_data = 13;
        Specified by:
        getExternalDataOrBuilder in interface OnnxMl.TensorProtoOrBuilder

      • getDataLocationValue

        public int getDataLocationValue()
         If value not set, data is stored in raw_data (if set) otherwise in type-specified field.
        .onnx.TensorProto.DataLocation data_location = 14;
        Specified by:
        getDataLocationValue in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        The enum numeric value on the wire for dataLocation.

      • getDoubleDataList

        public List<Double> getDoubleDataList()
         For double
 Complex128 tensors are encoded as a single array of doubles,
 with the real components appearing in odd numbered positions,
 and the corresponding imaginary component appearing in the
 subsequent even numbered position. (e.g., [1.0 + 2.0i, 3.0 + 4.0i]
 is encoded as [1.0, 2.0 ,3.0 ,4.0]
 When this field is present, the data_type field MUST be DOUBLE or COMPLEX128
        repeated double double_data = 10 [packed = true];
        Specified by:
        getDoubleDataList in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        A list containing the doubleData.

      • getDoubleDataCount

        public int getDoubleDataCount()
         For double
 Complex128 tensors are encoded as a single array of doubles,
 with the real components appearing in odd numbered positions,
 and the corresponding imaginary component appearing in the
 subsequent even numbered position. (e.g., [1.0 + 2.0i, 3.0 + 4.0i]
 is encoded as [1.0, 2.0 ,3.0 ,4.0]
 When this field is present, the data_type field MUST be DOUBLE or COMPLEX128
        repeated double double_data = 10 [packed = true];
        Specified by:
        getDoubleDataCount in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        The count of doubleData.

      • getDoubleData

        public double getDoubleData​(int index)
         For double
 Complex128 tensors are encoded as a single array of doubles,
 with the real components appearing in odd numbered positions,
 and the corresponding imaginary component appearing in the
 subsequent even numbered position. (e.g., [1.0 + 2.0i, 3.0 + 4.0i]
 is encoded as [1.0, 2.0 ,3.0 ,4.0]
 When this field is present, the data_type field MUST be DOUBLE or COMPLEX128
        repeated double double_data = 10 [packed = true];
        Specified by:
        getDoubleData in interface OnnxMl.TensorProtoOrBuilder
        Parameters:
        index - The index of the element to return.
        Returns:
        The doubleData at the given index.

      • getUint64DataList

        public List<Long> getUint64DataList()
         For uint64 and uint32 values
 When this field is present, the data_type field MUST be
 UINT32 or UINT64
        repeated uint64 uint64_data = 11 [packed = true];
        Specified by:
        getUint64DataList in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        A list containing the uint64Data.

      • getUint64DataCount

        public int getUint64DataCount()
         For uint64 and uint32 values
 When this field is present, the data_type field MUST be
 UINT32 or UINT64
        repeated uint64 uint64_data = 11 [packed = true];
        Specified by:
        getUint64DataCount in interface OnnxMl.TensorProtoOrBuilder
        Returns:
        The count of uint64Data.

      • getUint64Data

        public long getUint64Data​(int index)
         For uint64 and uint32 values
 When this field is present, the data_type field MUST be
 UINT32 or UINT64
        repeated uint64 uint64_data = 11 [packed = true];
        Specified by:
        getUint64Data in interface OnnxMl.TensorProtoOrBuilder
        Parameters:
        index - The index of the element to return.
        Returns:
        The uint64Data at the given index.

      • isInitialized

        public final boolean isInitialized()
        Specified by:
        isInitialized in interface org.nd4j.shade.protobuf.MessageLiteOrBuilder
        Overrides:
        isInitialized in class org.nd4j.shade.protobuf.GeneratedMessageV3

      • writeTo

        public void writeTo​(org.nd4j.shade.protobuf.CodedOutputStream output)
             throws IOException
        Specified by:
        writeTo in interface org.nd4j.shade.protobuf.MessageLite
        Overrides:
        writeTo in class org.nd4j.shade.protobuf.GeneratedMessageV3
        Throws:
        IOException

      • getSerializedSize

        public int getSerializedSize()
        Specified by:
        getSerializedSize in interface org.nd4j.shade.protobuf.MessageLite
        Overrides:
        getSerializedSize in class org.nd4j.shade.protobuf.GeneratedMessageV3

      • equals

        public boolean equals​(Object obj)
        Specified by:
        equals in interface org.nd4j.shade.protobuf.Message
        Overrides:
        equals in class org.nd4j.shade.protobuf.AbstractMessage

      • hashCode

        public int hashCode()
        Specified by:
        hashCode in interface org.nd4j.shade.protobuf.Message
        Overrides:
        hashCode in class org.nd4j.shade.protobuf.AbstractMessage

      • parseFrom

        public static OnnxMl.TensorProto parseFrom​(ByteBuffer data)
                                    throws org.nd4j.shade.protobuf.InvalidProtocolBufferException
        Throws:
        org.nd4j.shade.protobuf.InvalidProtocolBufferException

      • parseFrom

        public static OnnxMl.TensorProto parseFrom​(ByteBuffer data,
                                           org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry)
                                    throws org.nd4j.shade.protobuf.InvalidProtocolBufferException
        Throws:
        org.nd4j.shade.protobuf.InvalidProtocolBufferException

      • parseFrom

        public static OnnxMl.TensorProto parseFrom​(org.nd4j.shade.protobuf.ByteString data)
                                    throws org.nd4j.shade.protobuf.InvalidProtocolBufferException
        Throws:
        org.nd4j.shade.protobuf.InvalidProtocolBufferException

      • parseFrom

        public static OnnxMl.TensorProto parseFrom​(org.nd4j.shade.protobuf.ByteString data,
                                           org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry)
                                    throws org.nd4j.shade.protobuf.InvalidProtocolBufferException
        Throws:
        org.nd4j.shade.protobuf.InvalidProtocolBufferException

      • parseFrom

        public static OnnxMl.TensorProto parseFrom​(byte[] data)
                                    throws org.nd4j.shade.protobuf.InvalidProtocolBufferException
        Throws:
        org.nd4j.shade.protobuf.InvalidProtocolBufferException

      • parseFrom

        public static OnnxMl.TensorProto parseFrom​(byte[] data,
                                           org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry)
                                    throws org.nd4j.shade.protobuf.InvalidProtocolBufferException
        Throws:
        org.nd4j.shade.protobuf.InvalidProtocolBufferException

      • parseFrom

        public static OnnxMl.TensorProto parseFrom​(org.nd4j.shade.protobuf.CodedInputStream input,
                                           org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry)
                                    throws IOException
        Throws:
        IOException

      • newBuilderForType

        public OnnxMl.TensorProto.Builder newBuilderForType()
        Specified by:
        newBuilderForType in interface org.nd4j.shade.protobuf.Message
        Specified by:
        newBuilderForType in interface org.nd4j.shade.protobuf.MessageLite

      • toBuilder

        public OnnxMl.TensorProto.Builder toBuilder()
        Specified by:
        toBuilder in interface org.nd4j.shade.protobuf.Message
        Specified by:
        toBuilder in interface org.nd4j.shade.protobuf.MessageLite

      • newBuilderForType

        protected OnnxMl.TensorProto.Builder newBuilderForType​(org.nd4j.shade.protobuf.GeneratedMessageV3.BuilderParent parent)
        Specified by:
        newBuilderForType in class org.nd4j.shade.protobuf.GeneratedMessageV3

      • getParserForType

        public org.nd4j.shade.protobuf.Parser<OnnxMl.TensorProto> getParserForType()
        Specified by:
        getParserForType in interface org.nd4j.shade.protobuf.Message
        Specified by:
        getParserForType in interface org.nd4j.shade.protobuf.MessageLite
        Overrides:
        getParserForType in class org.nd4j.shade.protobuf.GeneratedMessageV3

      • getDefaultInstanceForType

        public OnnxMl.TensorProto getDefaultInstanceForType()
        Specified by:
        getDefaultInstanceForType in interface org.nd4j.shade.protobuf.MessageLiteOrBuilder
        Specified by:
        getDefaultInstanceForType in interface org.nd4j.shade.protobuf.MessageOrBuilder