org.bytedeco.pytorch.BatchNorm2dImpl Maven / Gradle / Ivy

Go to download
// Targeted by JavaCPP version 1.5.9: DO NOT EDIT THIS FILE

package org.bytedeco.pytorch;

import org.bytedeco.pytorch.Allocator;
import org.bytedeco.pytorch.Function;
import org.bytedeco.pytorch.Module;
import java.nio.*;
import org.bytedeco.javacpp.*;
import org.bytedeco.javacpp.annotation.*;

import static org.bytedeco.javacpp.presets.javacpp.*;
import static org.bytedeco.openblas.global.openblas_nolapack.*;
import static org.bytedeco.openblas.global.openblas.*;

import static org.bytedeco.pytorch.global.torch.*;


// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ BatchNorm2d
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

/** Applies the BatchNorm2d function.
 *  See https://pytorch.org/docs/master/nn.html#torch.nn.BatchNorm2d to learn
 *  about the exact behavior of this module.
 * 
 *  See the documentation for {@code torch::nn::BatchNorm2dOptions} class to learn
 *  what constructor arguments are supported for this module.
 * 
 *  Example:
 *  {@code
 *  BatchNorm2d
 *  model(BatchNorm2dOptions(4).eps(0.5).momentum(0.1).affine(false).track_running_stats(true));
 *  } */
@Namespace("torch::nn") @Properties(inherit = org.bytedeco.pytorch.presets.torch.class)
public class BatchNorm2dImpl extends BatchNorm2dImplBase {
    static { Loader.load(); }

  
    
      public BatchNorm2dImpl(@Const @ByRef BatchNormOptions options_) { super((Pointer)null); allocate(options_); }
      private native void allocate(@Const @ByRef BatchNormOptions options_);
    /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */
    public BatchNorm2dImpl(Pointer p) { super(p); }

}