Ejemplo n.º 1
0
import pyOcean_cpu as ocean

a = ocean.int8(0)
print(a)
a.imag = 3
print(a)
a.real = -1
print(a)
print(a.dtype)

a = ocean.float(3)
a.imag = ocean.float(2)
print(a)
print(a.dtype)
## Automatic type case and broadcast
import pyOcean_cpu as ocean
import sys

def exceptionMsg() :
   print("Expected error: %s" % str(sys.exc_info()[1]))

def failTest(command) :
   try :
      eval(command)
   except :
      exceptionMsg()


a = ocean.int16([1,2,3])
b = ocean.float([1])

print(a+b)

ocean.setAutoTypecast(False)
print(a+3)

failTest("a+3.2")

print(b+3)
print(b+3.2)
failTest("a+b")


ocean.setAutoBroadcast(False)
print(a+3)
Ejemplo n.º 3
0
import pyOcean_cpu as ocean

a = ocean.bool(0)
print(a.asPython())

a = ocean.int16(1)
print(a.asPython())

a = ocean.float(2)
print(a.asPython())

a = ocean.chalf(3 + 4j)
print(a.asPython())
Ejemplo n.º 4
0
import pyOcean_cpu as ocean

a = ocean.asTensor([[1, 2], [3, 4]], 'r', ocean.float)
b = ocean.arange(6, ocean.float).reshape([2, 3])

print(a)
print(b)
print(ocean.gemm(ocean.float(3), a, b))

alpha = ocean.asTensor([1, 2, 3], ocean.float).reshape([1, 1, 3])
print(ocean.gemm(alpha, a, b))
print(ocean.gemm(ocean.float(1), a.T, b))
Ejemplo n.º 5
0
import pyOcean_cpu as ocean

a = ocean.asTensor([[1 + 2j, 2 + 1j], [3 + 0j, 4 + 2j]], 'r', ocean.cfloat)
b = ocean.asTensor([1 + 1j, 2 - 3j], ocean.cfloat)

print(ocean.gemm(ocean.float(1), a, b))
Ejemplo n.º 6
0
import pyOcean_cpu as ocean

# Type casting applied to storage objects
s = ocean.asTensor([1, 2, 3]).storage
print(s)

t = ocean.int8(s)
print(t)

ocean.float(s, True)
print(s)
import pyOcean_cpu as ocean

a = ocean.float([1.0])
b = ocean.cdouble(1.0 + 0.01j)
c = ocean.cdouble(1.0 - 0.01j)

print([ocean.allLT(a, b), True])
print([ocean.allLT(a, c), False])
print([ocean.allGT(a, b), False])
print([ocean.allGE(a, c), True])
Ejemplo n.º 8
0
def failTest(command):
    try:
        eval(command)
    except:
        exceptionMsg()


a = ocean.int8(3)
print(a**2)
print(pow(a, 2))
a **= 2
print(a)

failTest("pow(a,2,3)")

print("\n------ Floating point ------")
a = ocean.float(4)
b = ocean.int16(3)
print(a**b)
print(pow(a, b))
a **= b
print(a)

print("\n------ Square root ------")
a = ocean.half(4)
b = a**ocean.half(0.5)
print([b, b.dtype.name])
print(a**0.5)
print(ocean.power(a, 0.5))
print(pow(a, 0.5))
Ejemplo n.º 9
0
import pyOcean_cpu as ocean

a = [-20, 20, -20, 20]
b = [3, 3, -3, -3]
c = [1, 2, -2, -1]
print(ocean.mod(ocean.int64(a), ocean.int64(b)))
print(ocean.mod(ocean.float(a), ocean.float(b)))
print(ocean.mod(ocean.double(a), ocean.double(b)))
print(ocean.int64(c))
print("")

c = [-2, 2, -2, 2]
print(ocean.fmod(ocean.int64(a), ocean.int64(b)))
print(ocean.fmod(ocean.float(a), ocean.float(b)))
print(ocean.fmod(ocean.double(a), ocean.double(b)))
print(ocean.int64(c))
Ejemplo n.º 10
0
import pyOcean_cpu as ocean

a = ocean.asTensor([[0, 1], [2, 3]], 'r', ocean.float)
b = ocean.asTensor([1, 2], ocean.float)
c = ocean.tensor([2, 1, 3], ocean.float)

ocean.gemm(ocean.float(1), a, b, 0, c)
print(c)
Ejemplo n.º 11
0
import pyOcean_cpu as ocean

# Type casting applied to tensor objects
a = ocean.asTensor([1, 2, 3])
print(a)

b = ocean.int8(a)
print(b)

ocean.float(a, True)
print(a)