FilesExpand file tree

 master

/

data.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

173 lines (133 loc) · 4.74 KB

 master

/

data.py

Top

File metadata and controls

• Code
• Blame

173 lines (133 loc) · 4.74 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

#######################################################

# Copyright (c) 2015, 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

########################################################

from sys import version_info

from .library import *

from .array import *

from .util import *

def constant(val, d0, d1=None, d2=None, d3=None, dtype=f32):

out = array()

out.arr = constant_array(val, d0, d1, d2, d3, dtype)

return out

# Store builtin range function to be used later

brange = range

def range(d0, d1=None, d2=None, d3=None, dim=-1, dtype=f32):

if not isinstance(dtype, ct.c_int):

if isinstance(dtype, int):

dtype = ct.c_int(dtype)

else:

raise TypeError("Invalid dtype")

out = array()

dims = dim4(d0, d1, d2, d3)

safe_call(clib.af_range(ct.pointer(out.arr), 4, ct.pointer(dims), dim, dtype))

return out

def iota(d0, d1=None, d2=None, d3=None, dim=-1, tile_dims=None, dtype=f32):

if not isinstance(dtype, ct.c_int):

if isinstance(dtype, int):

dtype = ct.c_int(dtype)

else:

raise TypeError("Invalid dtype")

out = array()

dims = dim4(d0, d1, d2, d3)

td=[1]*4

if tile_dims is not None:

for i in brange(len(tile_dims)):

td[i] = tile_dims[i]

tdims = dim4(td[0], td[1], td[2], td[3])

safe_call(clib.af_iota(ct.pointer(out.arr), 4, ct.pointer(dims), 4, ct.pointer(tdims), dtype))

return out

def randu(d0, d1=None, d2=None, d3=None, dtype=f32):

if not isinstance(dtype, ct.c_int):

if isinstance(dtype, int):

dtype = ct.c_int(dtype)

else:

raise TypeError("Invalid dtype")

out = array()

dims = dim4(d0, d1, d2, d3)

safe_call(clib.af_randu(ct.pointer(out.arr), 4, ct.pointer(dims), dtype))

return out

def randn(d0, d1=None, d2=None, d3=None, dtype=f32):

if not isinstance(dtype, ct.c_int):

if isinstance(dtype, int):

dtype = ct.c_int(dtype)

else:

raise TypeError("Invalid dtype")

out = array()

dims = dim4(d0, d1, d2, d3)

safe_call(clib.af_randn(ct.pointer(out.arr), 4, ct.pointer(dims), dtype))

return out

def set_seed(seed=0):

safe_call(clib.af_set_seed(ct.c_ulonglong(seed)))

def get_seed():

seed = ct.c_ulonglong(0)

safe_call(clib.af_get_seed(ct.pointer(seed)))

return seed.value

def identity(d0, d1=None, d2=None, d3=None, dtype=f32):

if not isinstance(dtype, ct.c_int):

if isinstance(dtype, int):

dtype = ct.c_int(dtype)

else:

raise TypeError("Invalid dtype")

out = array()

dims = dim4(d0, d1, d2, d3)

safe_call(clib.af_identity(ct.pointer(out.arr), 4, ct.pointer(dims), dtype))

return out

def diag(a, num=0, extract=True):

out = array()

if extract:

safe_call(clib.af_diag_extract(ct.pointer(out.arr), a.arr, ct.c_int(num)))

else:

safe_call(clib.af_diag_create(ct.pointer(out.arr), a.arr, ct.c_int(num)))

return out

def join(dim, first, second, third=None, fourth=None):

out = array()

if (third is None and fourth is None):

safe_call(clib.af_join(ct.pointer(out.arr), dim, first.arr, second.arr))

else:

ct.c_array_vec = dim4(first, second, 0, 0)

num = 2

if third is not None:

ct.c_array_vec[num] = third.arr

num+=1

if fourth is not None:

ct.c_array_vec[num] = fourth.arr

num+=1

safe_call(clib.af_join_many(ct.pointer(out.arr), dim, num, ct.pointer(ct.c_array_vec)))

def tile(a, d0, d1=1, d2=1, d3=1):

out = array()

safe_call(clib.af_tile(ct.pointer(out.arr), a.arr, d0, d1, d2, d3))

return out

def reorder(a, d0=1, d1=0, d2=2, d3=3):

out = array()

safe_call(clib.af_reorder(ct.pointer(out.arr), a.arr, d0, d1, d2, d3))

return out

def shift(a, d0, d1=0, d2=0, d3=0):

out = array()

safe_call(clib.af_shift(ct.pointer(out.arr), a.arr, d0, d1, d2, d3))

return out

def moddims(a, d0, d1=1, d2=1, d3=1):

out = array()

dims = dim4(d0, d1, d2, d3)

safe_call(clib.af_moddims(ct.pointer(out.arr), a.arr, 4, ct.pointer(dims)))

return out

def flat(a):

out = array()

safe_call(clib.af_flat(ct.pointer(out.arr), a.arr))

return out

def flip(a, dim=0):

out = array()

safe_call(clib.af_flip(ct.pointer(out.arr), a.arr, ct.c_int(dim)))

return out

def lower(a, is_unit_diag=False):

out = array()

safe_call(clib.af_lower(ct.pointer(out.arr), a.arr, is_unit_diag))

return out

def upper(a, is_unit_diag=False):

out = array()

safe_call(clib.af_upper(ct.pointer(out.arr), a.arr, is_unit_diag))

return out