1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
use tensor_rs::tensor::Tensor; use super::OpTrait; // BatchNorm1d // BatchNorm2d // BatchNorm3d // GroupNorm // SyncBatchNorm // InstanceNorm1d // InstanceNorm2d // InstanceNorm3d // LayerNorm // LocalResponseNorm