def test_long_tensor(): PyTorch.manualSeed(123) print('test_long_tensor') a = PyTorch.LongTensor(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_pytorchLong(): PyTorch.manualSeed(123) numpy.random.seed(123) LongTensor = PyTorch.LongTensor D = PyTorch.LongTensor(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.LongTensor(3, 4).bernoulli()) print(PyTorch.LongTensor(3, 4).geometric()) print(PyTorch.LongTensor(3, 4).geometric()) PyTorch.manualSeed(3) print(PyTorch.LongTensor(3, 4).geometric()) PyTorch.manualSeed(3) print(PyTorch.LongTensor(3, 4).geometric()) print(type(PyTorch.LongTensor(2, 3))) size = PyTorch.LongStorage(2) size[0] = 4 size[1] = 3 D.resize(size) print('D after resize:\n', D) print('resize1d', PyTorch.LongTensor().resize1d(3).fill(1)) print('resize2d', PyTorch.LongTensor().resize2d(2, 3).fill(1)) print('resize', PyTorch.LongTensor().resize(size).fill(1)) D = PyTorch.LongTensor(size).geometric() # def myeval(expr): # print(expr, ':', eval(expr)) # def myexec(expr): # print(expr) # exec(expr) myeval('LongTensor(3,2).nElement()') myeval('LongTensor().nElement()') myeval('LongTensor(1).nElement()') A = LongTensor(3, 4).geometric(0.9) myeval('A') myexec('A += 3') myeval('A') myexec('A *= 3') myeval('A') myexec('A -= 3') myeval('A') print('A //= 3') A //= 3 myeval('A') myeval('A + 5') myeval('A - 5') myeval('A * 5') print('A // 2') A // 2 B = LongTensor().resizeAs(A).geometric(0.9) myeval('B') myeval('A + B') myeval('A - B') myexec('A += B') myeval('A') myexec('A -= B') myeval('A')