Module math::tensor::tensor_shape

source ·

Structs

  • 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