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