How to experiment on test_torch.py?


(dl4daniel) #1

Hi Everyone,

@Veril @apaszke

I am new to pytorch, and I would like to learn more of tensor operations by experimenting the test_torch.py file, since it got lots of examples inside.

Basically, I want to run each test_function on each tensor operation individually, so I can see how the operation work.

How I tried, but failed:

# inside jupyter notebook
%cd  /path to /pytorch-master/test

import torch
import test_torch
test1 = test_torch.TestTorch(methodName='runTest')
test1.test_dot()

However, there is no error, nor return anything at all. Then I went to test_torch.py and added two print to test_dot() function as below, but nothing showed.

def test_dot(self):
        types = {
            'torch.DoubleTensor': 1e-8,
            'torch.FloatTensor': 1e-4,
        }
        for tname, prec in types.items():
            v1 = torch.randn(100).type(tname)
            v2 = torch.randn(100).type(tname)
            res1 = torch.dot(v1, v2)
            print(res1)
            res2 = 0
            for i, j in zip(v1, v2):
                res2 += i * j
            print(res2)
            self.assertEqual(res1, res2)

Could anyone help me find a way to experiment on each test function individually?

Thanks a lot!

Daniel