Module shape

Module shape 

Source
Expand description

Shape and stride handling for tensors

Structsยง

Shape
Shape of a tensor - dimensions along each axis
Strides
Strides for memory layout - bytes to skip for each dimension