Searched refs:permuted_tensor (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/test/ |
| D | stride_properties_test.cpp | 53 …auto permuted_tensor = at::rand({6, 3, 1, 5, 2}).permute({0, 3, 2, 1, 4}); // permute dim-1 & dim-3 in TEST() local
54 auto tensor = permuted_tensor.expand({6, 5, 4, 3, 2}); // expand dim-2 in TEST()
|