Function burn::tensor::ops::binary_ops_shape

pub fn binary_ops_shape(lhs: &[usize], rhs: &[usize]) -> Vec<usize>
Expand description

Computes the output shape for binary operations with broadcasting support.