(shape: ArrayShape, num_dims: usize)
| 81 | } |
| 82 | |
| 83 | pub fn prepend_dims(shape: ArrayShape, num_dims: usize) -> Result<ArrayShape> { |
| 84 | match shape.len() { |
| 85 | len if len == num_dims => Ok(shape), |
| 86 | len if len < num_dims => Ok([vec![1; num_dims - shape.len()], shape].concat()), |
| 87 | _ => Err(runtime_error!( |
| 88 | "prepend_dims(num_dims={num_dims}): input shape {shape:?} too large" |
| 89 | )), |
| 90 | } |
| 91 | } |
| 92 | |
| 93 | pub fn reshape_prepending_dims(node: Node, num_dims: usize) -> Result<Node> { |
| 94 | let t = node.get_type()?; |
no outgoing calls
no test coverage detected