pub fn element_count(shape: &[usize]) -> usize
Returns the total number of elements for a tensor with the given shape.