pytorch

1

2

3

4

5

6
import numpy as np
7

8
from caffe2.python import core, workspace
9
from caffe2.python.test_util import TestCase
10

11

12
class TestDuplicateOperands(TestCase):
13
    def test_duplicate_operands(self):
14
        net = core.Net('net')
15
        shape = (2, 4)
16
        x_in = np.random.uniform(size=shape)
17
        x = net.GivenTensorFill([], 'X', shape=shape,
18
                                values=x_in.flatten().tolist())
19
        xsq = net.Mul([x, x])
20
        y = net.DotProduct([xsq, xsq])
21
        net.AddGradientOperators([y])
22
        workspace.RunNetOnce(net)
23
        self.assertTrue(np.allclose(workspace.FetchBlob('X_grad'),
24
                                    4 * x_in**3))
25

26
if __name__ == "__main__":
27
    import unittest
28
    unittest.main()
29