def test_byte_tensor():
PyTorch.manualSeed(123)
print('test_byte_tensor')
a = PyTorch.ByteTensor(3, 2).geometric()
myeval('a')
myexec('a[1][1] = 9')
myeval('a')
myeval('a.size()')
myeval('a + 2')
myexec('a.resize2d(3,3).fill(1)')
myeval('a')
def test_pytorchByte():
PyTorch.manualSeed(123)
numpy.random.seed(123)
ByteTensor = PyTorch.ByteTensor
D = PyTorch.ByteTensor(5, 3).fill(1)
print('D', D)
D[2][2] = 4
print('D', D)
D[3].fill(9)
print('D', D)
D.narrow(1, 2, 1).fill(0)
print('D', D)
print(PyTorch.ByteTensor(3, 4).bernoulli())
print(PyTorch.ByteTensor(3, 4).geometric())
print(PyTorch.ByteTensor(3, 4).geometric())
PyTorch.manualSeed(3)
print(PyTorch.ByteTensor(3, 4).geometric())
PyTorch.manualSeed(3)
print(PyTorch.ByteTensor(3, 4).geometric())
print(type(PyTorch.ByteTensor(2, 3)))
size = PyTorch.LongStorage(2)
size[0] = 4
size[1] = 3
D.resize(size)
print('D after resize:\n', D)
print('resize1d', PyTorch.ByteTensor().resize1d(3).fill(1))
print('resize2d', PyTorch.ByteTensor().resize2d(2, 3).fill(1))
print('resize', PyTorch.ByteTensor().resize(size).fill(1))
D = PyTorch.ByteTensor(size).geometric()
# def myeval(expr):
# print(expr, ':', eval(expr))
# def myexec(expr):
# print(expr)
# exec(expr)
myeval('ByteTensor(3,2).nElement()')
myeval('ByteTensor().nElement()')
myeval('ByteTensor(1).nElement()')
A = ByteTensor(3, 4).geometric(0.9)
myeval('A')
myexec('A += 3')
myeval('A')
myexec('A *= 3')
myeval('A')
myeval('A')
print('A //= 3')
A //= 3
myeval('A')
myeval('A + 5')
myeval('A * 5')
print('A // 2')
A // 2
B = ByteTensor().resizeAs(A).geometric(0.9)
myeval('B')
myeval('A + B')
myexec('A += B')
myeval('A')
