Function size

gpu.hpp:86–92

* @brief Returns the number of elements in a tensor with the given shape, * which is equal to the product of the dimensions. * @param[in] shape Shape of the tensor * @return Number of elements in the tensor * * @code * size({256, 256}) -> 65536 * @endcode */

 * @endcode
 */
inline size_t size(const Shape &shape) {
  size_t numels = 1;
  for (size_t i = 0; i < shape.rank; i++) {
    numels *= shape.data[i];
  }
  return numels;
}

/**
 * @brief Represents a tensor on the GPU, which is a buffer of values with a