Struct SolveGrad 

pub struct SolveGrad<T: Scalar> {
    pub a: Tensor<T>,
    pub b: Tensor<T>,
}

Gradient result for solve_rrule: cotangents for both A and b.

Examples

use tenferro_device::LogicalMemorySpace;
use tenferro_linalg::solve_rrule;
use tenferro_prims::CpuContext;
use tenferro_tensor::{MemoryOrder, Tensor};

let mem = LogicalMemorySpace::MainMemory;
let col = MemoryOrder::ColumnMajor;
let mut ctx = CpuContext::new(1);
let a = Tensor::<f64>::eye(3, mem, col).unwrap();
let b = Tensor::<f64>::ones(&[3], mem, col).unwrap();
let cotangent = Tensor::<f64>::ones(&[3], mem, col).unwrap();
let grad = solve_rrule(&mut ctx, &a, &b, &cotangent).unwrap();
assert_eq!(grad.a.dims(), &[3, 3]);
assert_eq!(grad.b.dims(), &[3]);

Fields

a: Tensor<T>

Cotangent for the coefficient matrix A.

b: Tensor<T>

Cotangent for the right-hand side b.

Trait Implementations

impl<T: Debug + Scalar> Debug for SolveGrad<T>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.