oa::ml::matrix::channel_norm_relu

Normalize the channel axis of [B, C, T] and fuse the following ReLU. Errors Returns an error under the same conditions as channel_norm, or when runtime recording fails.

FnRust public API

Function Documentation

pub fn channel_norm_relu(
	input: &Matrix,
	weight: &Matrix,
	bias: &Matrix,
	epsilon: f32
) -> Result<Matrix>

Normalize the channel axis of [B, C, T] and fuse the following ReLU. Errors Returns an error under the same conditions as channel_norm, or when runtime recording fails.