xref: /aosp_15_r20/external/pytorch/test/bottleneck_test/test.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1# Owner(s): ["module: unknown"]
2
3import torch
4
5
6x = torch.ones((3, 3), requires_grad=True)
7(3 * x).sum().backward()
8