Repository navigation
Expand file tree
/
Copy pathinteger.lua
More file actions
71 lines (68 loc) · 1.82 KB
/
Copy pathinteger.lua
File metadata and controls
71 lines (68 loc) · 1.82 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
--[[
/*******************************************************
* Copyright (c) 2014, ArrayFire
* All rights reserved.
*
* This file is distributed under 3-clause BSD license.
* The complete license agreement can be obtained at:
* http://arrayfire.com/licenses/BSD-3-Clause
********************************************************/
]]
-- Modules --
local AF = require("arrayfire")
print("")
print("N.B. THIS IS A WIP")
print("")
AF.main(function()
print("\n=== ArrayFire signed(s32) / unsigned(u32) Integer Example ===")
local h_A = {1, 2, 4, -1, 2, 0, 4, 2, 3}
local h_B = {2, 3, -5, 6, 0, 10, -12, 0, 1}
local A = AF.array(3, 3, h_A)
local B = AF.array(3, 3, h_B)
print("--\nSub-refencing and Sub-assignment")
AF.print("A", A)
-- af_print(A.col(0))
-- af_print(A.row(0))
-- A(0) = 11
-- A(1) = 100
AF.print("A", A)
AF.print("B", B)
-- A(1,span) = B(2,span);
AF.print("A", A)
--[[
print("--Bit-wise operations")
-- Returns an array of type s32
af_print(A & B)
af_print(A | B)
af_print(A ^ B)
print("\n--Logical operations")
-- Returns an array of type b8
af_print(A && B)
af_print(A || B)
]]
print("\n--Transpose")
AF.print("A", A)
AF.print("A.T()", A:T())
print("\n--Flip Vertically / Horizontally")
AF.print("A", A)
AF.print("flip(A,0)", AF.flip(A,0))
AF.print("flip(A,1)", AF.flip(A,1))
print("\n--Sum along columns")
AF.print("A", A)
AF.print("sum(A)", AF.sum(A))
print("\n--Product along columns")
AF.print("A", A)
AF.print("product(A)", AF.product(A))
print("\n--Minimum along columns")
AF.print("A", A)
AF.print("min(A)", AF.min(A))
print("\n--Maximum along columns")
AF.print("A", A)
AF.print("max(A)", AF.max(A))
print("\n--Minimum along columns with index")
AF.print("A", A)
local out, idx = AF.array(), AF.array()
AF.min(out, idx, A)
AF.print("out", out)
AF.print("idx", idx)
end)