You can use torch::sizes() method IntArrayRef sizes(). It's equivalent of shape in python. Furthermore you can access specific size at given ...
確定! 回上一頁