SwiftNP is a Swift package for numerical computing inspired by the popular Python library, NumPy. It provides support for multi-dimensional arrays (NDArray), along with basic operations on these arrays, such as reshaping, element-wise operations, and more.
- Multi-dimensional arrays (
NDArray
) - Support for various numeric types (e.g.,
Int
,Float
,Double
, etc.) - Basic array operations (reshape, element-wise operations, etc.)
- Extensible architecture for additional numerical computing functionality
Add the following to your Package.swift
file:
dependencies: [
.package(url: "https://github.com/yourusername/SwiftNP", from: "0.0.1")
]
Then, include SwiftNP as a dependency in your target:
.target(
name: "YourTarget",
dependencies: ["SwiftNP"]),
Import SwiftNP
In any Swift file where you want to use SwiftNP:
import SwiftNP
You can create a multi-dimensional array by specifying the shape and default values:
let array = SNP.ones(shape: [1, 3, 512, 512])
SwiftNP supports element-wise operations, reshaping, and much more:
let reshapedArray = array.reshape([1, 9])
SwiftNP supports a wide range of numeric data types. You can specify the data type when creating an NDArray:
let floatArray = NDArray(shape: [2, 2], dtype: .float64, defaultValue: 21.0)
This project is licensed under the MIT License. See the LICENSE file for details.