pub struct GroupNorm {
pub ws: Option<Tensor>,
pub bs: Option<Tensor>,
pub num_groups: i64,
pub num_channels: i64,
/* private fields */
}
Expand description
A group-normalization layer.
Fields§
§ws: Option<Tensor>
§bs: Option<Tensor>
§num_groups: i64
§num_channels: i64