(node: Node, num_dims: usize)
| 91 | } |
| 92 | |
| 93 | pub fn reshape_prepending_dims(node: Node, num_dims: usize) -> Result<Node> { |
| 94 | let t = node.get_type()?; |
| 95 | let shape = t.get_shape(); |
| 96 | let new_shape = prepend_dims(shape.clone(), num_dims)?; |
| 97 | if shape == new_shape { |
| 98 | Ok(node) |
| 99 | } else { |
| 100 | Ok(node.reshape(array_type(new_shape, t.get_scalar_type()))?) |
| 101 | } |
| 102 | } |
| 103 | |
| 104 | /// Panics if `x` is not an array. |
| 105 | pub fn put_in_bits(x: Node) -> Result<Node> { |
no test coverage detected