Searched refs:get_castable_tensor (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/test/ |
| D | test_numpy_interop.py | 50 def get_castable_tensor(shape, dtype): function
80 x = get_castable_tensor(sz, dtp)
86 xm = get_castable_tensor(sz * 2, dtp)
106 x = get_castable_tensor((sz1, sz2), dtp)
112 xm = get_castable_tensor((sz1 * 2, sz2), dtp)
120 x = get_castable_tensor((sz2, sz1), dtp).t()
126 xm = get_castable_tensor((sz2 * 2, sz1), dtp)
133 xm = get_castable_tensor((sz2 * 2, sz1 * 2), dtp)
141 x = get_castable_tensor((3, 4), dtp)
|