Skip to main content

integer_tensor_dimension_vector

Function integer_tensor_dimension_vector 

Source
pub fn integer_tensor_dimension_vector(
    tensor: &Tensor,
    name: &str,
    allow_zero: bool,
) -> Option<Result<Vec<usize>, String>>
Expand description

Parse every element of a typed integer tensor as a dimension selector without routing through the tensor’s f64 compatibility backing store.