org.bytedeco.pytorch

Class CrossEntropyLossImpl

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.pytorch.Module

org.bytedeco.pytorch.CrossEntropyLossImplCloneable

org.bytedeco.pytorch.CrossEntropyLossImpl

All Implemented Interfaces:

AutoCloseable

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

Creates a criterion that computes cross entropy loss between input and target. See https://pytorch.org/docs/master/nn.html#torch.nn.CrossEntropyLoss to learn about the exact behavior of this module. See the documentation for torch::nn::CrossEntropyLossOptions class to learn what constructor arguments are supported for this module. Example:

  CrossEntropyLoss
  model(CrossEntropyLossOptions().ignore_index(-100).reduction(torch::kMean));
  

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
CrossEntropyLossImpl()
CrossEntropyLossImpl(CrossEntropyLossOptions options_)
CrossEntropyLossImpl(Module pointer) Downcast constructor.
CrossEntropyLossImpl(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
Tensor	forward(Tensor input, Tensor target)
CrossEntropyLossOptions	options() The options with which this Module was constructed.
CrossEntropyLossImpl	options(CrossEntropyLossOptions setter)
void	pretty_print(Pointer stream) Pretty prints the CrossEntropyLoss module into the given stream.
void	reset() reset() must perform initialization of all members with reference semantics, most importantly parameters, buffers and submodules.
Tensor	weight() A manual rescaling weight given to to each class.
CrossEntropyLossImpl	weight(Tensor setter)

Methods inherited from class org.bytedeco.pytorch.CrossEntropyLossImplCloneable

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

CrossEntropyLossImpl

public CrossEntropyLossImpl(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

CrossEntropyLossImpl

public CrossEntropyLossImpl(Module pointer)

Downcast constructor.

CrossEntropyLossImpl

public CrossEntropyLossImpl(@ByVal(nullValue="torch::nn::CrossEntropyLossOptions{}")
                            CrossEntropyLossOptions options_)

CrossEntropyLossImpl

public CrossEntropyLossImpl()

Method Detail

reset

public void reset()

Description copied from class: CrossEntropyLossImplCloneable

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

Overrides:

reset in class CrossEntropyLossImplCloneable

pretty_print

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

Pretty prints the CrossEntropyLoss 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 CrossEntropyLossOptions options()

The options with which this Module was constructed.

options

public CrossEntropyLossImpl options(CrossEntropyLossOptions setter)

weight

@ByRef
public Tensor weight()

A manual rescaling weight given to to each class.

weight

public CrossEntropyLossImpl weight(Tensor setter)