it is a 4 dimensional tensor, but I would like to know how to do it for a tensor of arbitrary size.
Basically what I’m looking for is an output similar to
std::cout << x << '\n';

if I have to implement a python extension, so be it… but I still don’t understand how to do it.