org.bytedeco.pytorch

Class HuberLossImpl

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.pytorch.Module

org.bytedeco.pytorch.HuberLossImplCloneable

org.bytedeco.pytorch.HuberLossImpl

All Implemented Interfaces:

AutoCloseable

@Namespace(value="torch::nn")
 @NoOffset
 @Properties(inherit=torch.class)
public class HuberLossImpl
extends HuberLossImplCloneable

Creates a criterion that uses a squared term if the absolute element-wise error falls below delta and a delta-scaled L1 term otherwise. See https://pytorch.org/docs/master/nn.html#torch.nn.HuberLoss to learn about the exact behavior of this module. See the documentation for torch::nn::HuberLossOptions class to learn what constructor arguments are supported for this module. Example:

  HuberLoss model(HuberLossOptions().reduction(torch::kNone).delta(0.5));
  

Nested Class Summary

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter

Field Summary

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
HuberLossImpl()
HuberLossImpl(HuberLossOptions options_)
HuberLossImpl(Module pointer) Downcast constructor.
HuberLossImpl(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
Tensor	forward(Tensor input, Tensor target)
HuberLossOptions	options() The options with which this Module was constructed.
HuberLossImpl	options(HuberLossOptions setter)
void	pretty_print(Pointer stream) Pretty prints the HuberLoss module into the given stream.
void	reset() reset() must perform initialization of all members with reference semantics, most importantly parameters, buffers and submodules.

Methods inherited from class org.bytedeco.pytorch.HuberLossImplCloneable

asModule, asModule, clone, clone

Methods inherited from class org.bytedeco.pytorch.Module

apply, apply, apply, apply, apply, apply, apply, apply, buffers, buffers, children, eval, is_serializable, is_training, load, modules, modules, name, named_buffers, named_buffers, named_children, named_modules, named_modules, named_modules, named_parameters, named_parameters, parameters, parameters, register_buffer, register_buffer, register_module, register_module, register_parameter, register_parameter, register_parameter, register_parameter, save, shiftLeft, to, to, to, train, unregister_module, unregister_module, zero_grad

Methods inherited from class org.bytedeco.javacpp.Pointer

address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero

Methods inherited from class java.lang.Object

finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

HuberLossImpl

public HuberLossImpl(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

HuberLossImpl

public HuberLossImpl(Module pointer)

Downcast constructor.

HuberLossImpl

public HuberLossImpl(@ByVal(nullValue="torch::nn::HuberLossOptions{}")
                     HuberLossOptions options_)

HuberLossImpl

public HuberLossImpl()

Method Detail

reset

public void reset()

Description copied from class: HuberLossImplCloneable

reset() must perform initialization of all members with reference semantics, most importantly parameters, buffers and submodules.

Overrides:

reset in class HuberLossImplCloneable

pretty_print

public void pretty_print(@Cast(value="std::ostream*") @ByRef
                         Pointer stream)

Pretty prints the HuberLoss module into the given stream.

Overrides:

pretty_print in class Module

forward

@ByVal
public Tensor forward(@Const @ByRef
                             Tensor input,
                             @Const @ByRef
                             Tensor target)

options

@ByRef
public HuberLossOptions options()

The options with which this Module was constructed.

options

public HuberLossImpl options(HuberLossOptions setter)