Module tensor_shape

Source

Structs§

TensorShape
The shape of an N-dimensional tensor has a size for each dimension, with an associated stride, e.g., a row-major 3 x 5 matrix will have a stride of 5 for the dimension of size 3 and a stride of 1 for the dimension of size 5, and the resulting dims_strides is [(3, 5), (5, 1)]. Index 0 of dims_strides always refers to the leftmost dimension.

Traits§

HasTensorShape