Trait burn::tensor::BroadcastArgs
pub trait BroadcastArgs<const D1: usize, const D2: usize> {
// Required method
fn into_shape(self, shape: &Shape<D1>) -> Shape<D2>;
}
Expand description
Trait used for broadcast arguments.
Required Methods§
fn into_shape(self, shape: &Shape<D1>) -> Shape<D2>
fn into_shape(self, shape: &Shape<D1>) -> Shape<D2>
Converts to a shape.