oarust 0.8.6

GPU-first semantic computing engine written in Rust
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

//! Zero-copy Matrix views and their autograd lineage.

use crate::{Matrix, Result, matrix::autograd};

/// Return a zero-copy dense view with a different shape.
///
/// The result owns a new semantic value identity but shares storage, pending
/// execution state, dtype, and element order with `input`. When a gradient tape
/// is active, its adjoint reshapes the incoming gradient back to the input shape.
///
/// # Errors
///
/// Returns an error when the requested shape overflows or changes the number of
/// elements.
pub fn reshape(input: &Matrix, shape: impl Into<Vec<usize>>) -> Result<Matrix> {
	let output = input.reshape_view(shape.into())?;
	autograd::record_reshape(input, &output)?;
	Ok(output)
}

pub(crate) fn reshape_semantic_output(input: &Matrix, shape: Vec<usize>) -> Result<Matrix> {
	let output = input.reshape_semantic_output(shape)?;
	autograd::record_reshape(input, &output)?;
	Ok(output)
}

/// Restore the source shape without copying device storage.
pub(crate) fn reshape_backward(output_gradient: &Matrix, input_shape: &[usize]) -> Result<Matrix> {
	output_gradient.reshape(input_shape.to_vec())
}