MCPcopy Create free account
hub / github.com/NVIDIA/DALI / py_shape_list

Function py_shape_list

dali/python/backend_impl.cc:146–152  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

144
145template <typename Backend>
146std::vector<py::tuple> py_shape_list(const TensorList<Backend> &tl) {
147 std::vector<py::tuple> ret(tl.shape().size());
148 for (int i = 0; i < tl.shape().size(); ++i) {
149 ret[i] = py::tuple(as_py_list(tl.tensor_shape(i)));
150 }
151 return ret;
152}
153
154static string TensorLayoutRepr(const TensorLayout &tl) {
155 std::stringstream ss;

Callers

nothing calls this directly

Calls 4

as_py_listFunction · 0.85
sizeMethod · 0.45
shapeMethod · 0.45
tensor_shapeMethod · 0.45

Tested by

no test coverage detected