zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

WIP (examples)

Browse Source
This commit is contained in:
Ronan Collobert
2025-10-31 16:23:51 -07:00
parent c5913131cf
commit 7107802e09
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/cpp/tutorial.cpp
View File

@@ -14,14 +14,17 @@ void array_basics() {
// Get the value out of it: // Get the value out of it:
auto s = x.item<float>(); auto s = x.item<float>();
assert(s == 1.0); assert(s == 1.0);
(void)s;
// Scalars have a size of 1: // Scalars have a size of 1:
size_t size = x.size(); int64_t size = x.size();
assert(size == 1); assert(size == 1);
(void)size;
// Scalars have 0 dimensions: // Scalars have 0 dimensions:
int ndim = x.ndim(); int ndim = x.ndim();
assert(ndim == 0); assert(ndim == 0);
(void)ndim;
// The shape should be an empty vector: // The shape should be an empty vector:
auto shape = x.shape(); auto shape = x.shape();
@@ -30,6 +33,7 @@ void array_basics() {
// The datatype should be float32: // The datatype should be float32:
auto dtype = x.dtype(); auto dtype = x.dtype();
assert(dtype == mx::float32); assert(dtype == mx::float32);
(void)dtype;
// Specify the dtype when constructing the array: // Specify the dtype when constructing the array:
x = mx::array(1, mx::int32); x = mx::array(1, mx::int32);