Overloads for data Tensor constructor. (#9691)

Summary: #8366

Reviewed By: bsoyluoglu

Differential Revision: D71929194

Co-authored-by: Anthony Shoumikhin <[email protected]>

Author: kirklandsign

extension/apple/ExecuTorch/Exported/ExecuTorchTensor.h

@@ -367,6 +367,48 @@ __attribute__((deprecated("This API is experimental.")))
                     dataType:(ExecuTorchDataType)dataType
                shapeDynamism:(ExecuTorchShapeDynamism)shapeDynamism;
 
+/**
+ * Initializes a tensor using an NSData object as the underlying data buffer with dynamic bound shape.
+ *
+ * @param data An NSData object containing the tensor data.
+ * @param shape An NSArray of NSNumber objects representing the tensor's shape.
+ * @param strides An NSArray of NSNumber objects representing the tensor's strides.
+ * @param dimensionOrder An NSArray of NSNumber objects indicating the order of dimensions.
+ * @param dataType An ExecuTorchDataType value specifying the element type.
+ * @return An initialized ExecuTorchTensor instance using the provided data.
+ */
+- (instancetype)initWithData:(NSData *)data
+                       shape:(NSArray<NSNumber *> *)shape
+                     strides:(NSArray<NSNumber *> *)strides
+              dimensionOrder:(NSArray<NSNumber *> *)dimensionOrder
+                    dataType:(ExecuTorchDataType)dataType;
+
+/**
+ * Initializes a tensor using an NSData object as the underlying data buffer, specifying shape, data type, and explicit shape dynamism.
+ *
+ * @param data An NSData object containing the tensor data.
+ * @param shape An NSArray of NSNumber objects representing the tensor's shape.
+ * @param dataType An ExecuTorchDataType value specifying the element type.
+ * @param shapeDynamism An ExecuTorchShapeDynamism value indicating the shape dynamism.
+ * @return An initialized ExecuTorchTensor instance using the provided data.
+ */
+- (instancetype)initWithData:(NSData *)data
+                       shape:(NSArray<NSNumber *> *)shape
+                    dataType:(ExecuTorchDataType)dataType
+               shapeDynamism:(ExecuTorchShapeDynamism)shapeDynamism;
+
+/**
+ * Initializes a tensor using an NSData object as the underlying data buffer, specifying only the shape and data type.
+ *
+ * @param data An NSData object containing the tensor data.
+ * @param shape An NSArray of NSNumber objects representing the tensor's shape.
+ * @param dataType An ExecuTorchDataType value specifying the element type.
+ * @return An initialized ExecuTorchTensor instance using the provided data.
+ */
+- (instancetype)initWithData:(NSData *)data
+                       shape:(NSArray<NSNumber *> *)shape
+                    dataType:(ExecuTorchDataType)dataType;
+
 @end
 
 NS_ASSUME_NONNULL_END

extension/apple/ExecuTorch/Exported/ExecuTorchTensor.mm

@@ -298,4 +298,40 @@ - (instancetype)initWithData:(NSData *)data
   return self;
 }
 
+- (instancetype)initWithData:(NSData *)data
+                       shape:(NSArray<NSNumber *> *)shape
+                     strides:(NSArray<NSNumber *> *)strides
+              dimensionOrder:(NSArray<NSNumber *> *)dimensionOrder
+                    dataType:(ExecuTorchDataType)dataType {
+  return [self initWithData:data
+                      shape:shape
+                    strides:strides
+             dimensionOrder:dimensionOrder
+                   dataType:dataType
+              shapeDynamism:ExecuTorchShapeDynamismDynamicBound];
+}
+
+- (instancetype)initWithData:(NSData *)data
+                       shape:(NSArray<NSNumber *> *)shape
+                    dataType:(ExecuTorchDataType)dataType
+               shapeDynamism:(ExecuTorchShapeDynamism)shapeDynamism {
+  return [self initWithData:data
+                      shape:shape
+                    strides:@[]
+             dimensionOrder:@[]
+                   dataType:dataType
+              shapeDynamism:shapeDynamism];
+}
+
+- (instancetype)initWithData:(NSData *)data
+                       shape:(NSArray<NSNumber *> *)shape
+                    dataType:(ExecuTorchDataType)dataType {
+  return [self initWithData:data
+                      shape:shape
+                    strides:@[]
+             dimensionOrder:@[]
+                   dataType:dataType
+              shapeDynamism:ExecuTorchShapeDynamismDynamicBound];
+}
+
 @end

extension/apple/ExecuTorch/__tests__/TensorTest.swift

@@ -104,7 +104,7 @@ class TensorTest: XCTestCase {
   func testInitData() {
     let dataArray: [Float] = [1.0, 2.0, 3.0, 4.0]
     let data = Data(bytes: dataArray, count: dataArray.count * MemoryLayout<Float>.size)
-    let tensor = Tensor(data: data, shape: [4], strides: [1], dimensionOrder: [0], dataType: .float, shapeDynamism: .static)
+    let tensor = Tensor(data: data, shape: [4], dataType: .float)
     XCTAssertEqual(tensor.count, 4)
     tensor.bytes { pointer, count, dataType in
       XCTAssertEqual(Array(UnsafeBufferPointer(start: pointer.assumingMemoryBound(to: Float.self), count: count)), dataArray)