org.bytedeco.pytorch

Class TensorIteratorBase

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.pytorch.MetaBase

org.bytedeco.pytorch.TensorIteratorBase

All Implemented Interfaces:

AutoCloseable

Direct Known Subclasses:

TensorIterator

@Namespace(value="at")
 @NoOffset
 @Properties(inherit=torch.class)
public class TensorIteratorBase
extends MetaBase

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
TensorIteratorBase(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
void	_unsafe_set_arg_data(int arg, Pointer data)
void	_unsafe_set_arg_strides(int arg, long... strides)
void	_unsafe_set_arg_strides(int arg, LongArrayRef strides)
void	build_binary_float_op(TensorBase out, TensorBase a, TensorBase b)
void	build_binary_op(TensorBase out, TensorBase a, TensorBase b)
void	build_borrowing_binary_float_op(TensorBase out, TensorBase a, TensorBase b)
void	build_borrowing_binary_op(TensorBase out, TensorBase a, TensorBase b)
void	build_borrowing_comparison_op(TensorBase out, TensorBase a, TensorBase b)
void	build_borrowing_except_last_argument_comparison_op(TensorBase out, TensorBase a, TensorBase b)
void	build_borrowing_unary_float_op(TensorBase out, TensorBase a)
void	build_borrowing_unary_force_boolean_op(TensorBase out, TensorBase a)
void	build_borrowing_unary_op(TensorBase out, TensorBase a)
void	build_comparison_op(TensorBase out, TensorBase a, TensorBase b)
void	build_output_borrowing_argument_owning_unary_op(TensorBase out, TensorBase a)
void	build_ternary_op(TensorBase out, TensorBase a, TensorBase b, TensorBase c)
void	build_unary_float_op(TensorBase out, TensorBase a)
void	build_unary_op(TensorBase out, TensorBase a)
void	build(TensorIteratorConfig arg0)
boolean	can_use_32bit_indexing() true if the stride computation can use 32-bit arithmetic.
void	cast_outputs()
torch.ScalarType	common_dtype()
SymDimVector	compatible_stride(int element_size) Create a strides array for a Tensor with shape of this iterator.
Pointer	data_ptr(int arg)
torch.DeviceType	device_type()
torch.DeviceType	device_type(int arg)
Device	device()
Device	device(int arg)
torch.ScalarType	dtype()
torch.ScalarType	dtype(int arg)
long	element_size(int arg)
SymDimVector	get_base_ptrs()
SymDimVector	get_dim_strides(int dim) Helper functions for CPU iteration
int	get_dim_to_split() Returns the dimension with the largest extent: (size[dim]-1) * stride[dim]
SymDimVector	get_inner_strides()
SymDimVector	get_strides()
boolean	has_contiguous_first_dim()
TensorBase	input_base()
TensorBase	input_base(int arg)
torch.ScalarType	input_dtype()
torch.ScalarType	input_dtype(int arg)
Tensor	input()
Tensor	input(int arg)
DimVector	invert_perm(long... input)
DimVector	invert_perm(LongArrayRef input) Inverts the re-ordering done by reorder_dimensions.
boolean	is_contiguous() Reducible to 1-dimensional and all operands are contiguous
boolean	is_cpu_scalar(int arg)
boolean	is_dim_reduced(int dim)
boolean	is_final_output() Whether this iterator produces the actual output, as opposed to something that will be accumulated further.
boolean	is_scalar(int arg)
boolean	is_trivial_1d() 1-dimensional iteration and no buffering or type conversion
void	narrow(int dim, long start, long size) Shrinks an iterated dimension
int	ndim()
int	ninputs()
int	noutputs()
int	ntensors()
long	num_output_elements() number of elements in the output operand.
int	num_reduce_dims() number of reduced dimensions in a reduction operation
long	numel()
TensorBase	output_base()
TensorBase	output_base(int arg)
Tensor	output()
Tensor	output(int arg)
void	remove_operand(int arg) Removes an operand from this iterator
void	select_all_keeping_dim(int start_dim, long... starts)
void	select_all_keeping_dim(int start_dim, LongArrayRef starts) Narrows every dim after and including start_dim to size one.
void	set_output_raw_strided(long output_idx, long[] sizes, long[] strides, TensorOptions options, DimnameVector names)
void	set_output_raw_strided(long output_idx, LongArrayRef sizes, LongArrayRef strides, TensorOptions options, DimnameArrayRef names)
LongArrayRef	shape()
boolean	should_accumulate() If the kernel should accumulate into the output.
TensorIterator	split(int dim) Splits this TensorIterator into two iterators.
LongArrayRef	strides(int arg) Accessors for each operand
TensorBase	tensor_base(int arg)
Tensor	tensor(int arg)
void	unsafe_replace_operand(int arg, Pointer data) Replaces the data pointer for the operand at index arg.
LongArrayRef	view_offsets()
SplitUntil32Bit	with_32bit_indexing() An "iteratable" object that recursively splits this iterator into sub-iterators that can use 32-bit indexing.

Methods inherited from class org.bytedeco.pytorch.MetaBase

maybe_get_output, maybe_get_output, set_output_contiguous, set_output_contiguous, set_output_contiguous, set_output_contiguous, set_output_raw_strided, set_output_raw_strided, set_output_strided, set_output_strided, set_output_strided, set_output_strided

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

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

Constructor Detail

TensorIteratorBase

public TensorIteratorBase(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

Method Detail

build

public void build(@ByRef
                  TensorIteratorConfig arg0)

ndim

public int ndim()

shape

@ByVal
public LongArrayRef shape()

numel

@Cast(value="int64_t")
public long numel()

ntensors

public int ntensors()

noutputs

public int noutputs()

ninputs

public int ninputs()

view_offsets

@ByVal
public LongArrayRef view_offsets()

num_output_elements

@Cast(value="int64_t")
public long num_output_elements()

number of elements in the output operand. this is the same as numel() for operations that are not reductions.

num_reduce_dims

public int num_reduce_dims()

number of reduced dimensions in a reduction operation

is_trivial_1d

@Cast(value="bool")
public boolean is_trivial_1d()

1-dimensional iteration and no buffering or type conversion

is_contiguous

@Cast(value="bool")
public boolean is_contiguous()

Reducible to 1-dimensional and all operands are contiguous

is_dim_reduced

@Cast(value="bool")
public boolean is_dim_reduced(int dim)

strides

@ByVal
public LongArrayRef strides(int arg)

Accessors for each operand

data_ptr

public Pointer data_ptr(int arg)

dtype

public torch.ScalarType dtype(int arg)

dtype

public torch.ScalarType dtype()

common_dtype

public torch.ScalarType common_dtype()

input_dtype

public torch.ScalarType input_dtype(int arg)

input_dtype

public torch.ScalarType input_dtype()

device

@ByVal
public Device device(int arg)

device

@ByVal
public Device device()

device_type

public torch.DeviceType device_type(int arg)

device_type

public torch.DeviceType device_type()

element_size

@Cast(value="int64_t")
public long element_size(int arg)

is_scalar

@Cast(value="bool")
public boolean is_scalar(int arg)

is_cpu_scalar

@Cast(value="bool")
public boolean is_cpu_scalar(int arg)

tensor_base

@Const
 @ByRef
public TensorBase tensor_base(int arg)

tensor

@Const
 @ByRef
public Tensor tensor(int arg)

output_base

@Const
 @ByRef
public TensorBase output_base(int arg)

output_base

@Const
 @ByRef
public TensorBase output_base()

output

@Const
 @ByRef
public Tensor output(int arg)

output

@Const
 @ByRef
public Tensor output()

input_base

@Const
 @ByRef
public TensorBase input_base(int arg)

input_base

@Const
 @ByRef
public TensorBase input_base()

input

@Const
 @ByRef
public Tensor input(int arg)

input

@Const
 @ByRef
public Tensor input()

cast_outputs

public void cast_outputs()

remove_operand

public void remove_operand(int arg)

Removes an operand from this iterator

narrow

public void narrow(int dim,
                   @Cast(value="int64_t")
                   long start,
                   @Cast(value="int64_t")
                   long size)

Shrinks an iterated dimension

select_all_keeping_dim

public void select_all_keeping_dim(int start_dim,
                                   @ByVal
                                   LongArrayRef starts)

Narrows every dim after and including start_dim to size one.

select_all_keeping_dim

public void select_all_keeping_dim(int start_dim,
                                   @ByVal @Cast(value={"int64_t*","c10::ArrayRef<int64_t>","std::vector<int64_t>&"}) @StdVector(value="int64_t")
                                   long... starts)

unsafe_replace_operand

public void unsafe_replace_operand(int arg,
                                   Pointer data)

Replaces the data pointer for the operand at index arg. The new pointer should have the same sizes, strides and dtype as the original

split

@UniquePtr
 @ByVal
public TensorIterator split(int dim)

Splits this TensorIterator into two iterators. Together they iterate over the entire operation. Used by with_32bit_indexing().

get_dim_to_split

public int get_dim_to_split()

Returns the dimension with the largest extent: (size[dim]-1) * stride[dim]

compatible_stride

@ByVal
 @Cast(value="at::TensorIteratorBase::StrideVector*")
public SymDimVector compatible_stride(int element_size)

Create a strides array for a Tensor with shape of this iterator. The parameter element_size specifies the size of Tensor's data type in bytes (e.g. 4 for float)

invert_perm

@ByVal
public DimVector invert_perm(@ByVal
                                    LongArrayRef input)

Inverts the re-ordering done by reorder_dimensions. This can only be called *before* coalesce_dimensions() is called.

invert_perm

@ByVal
public DimVector invert_perm(@ByVal @Cast(value={"int64_t*","c10::ArrayRef<int64_t>","std::vector<int64_t>&"}) @StdVector(value="int64_t")
                                    long... input)

get_dim_strides

@ByVal
 @Cast(value="at::TensorIteratorBase::StrideVector*")
public SymDimVector get_dim_strides(int dim)

Helper functions for CPU iteration

get_strides

@ByVal
 @Cast(value="at::TensorIteratorBase::StrideVector*")
public SymDimVector get_strides()

get_inner_strides

@ByVal
 @Cast(value="at::TensorIteratorBase::StrideVector*")
public SymDimVector get_inner_strides()

get_base_ptrs

@ByVal
 @Cast(value="at::TensorIteratorBase::PtrVector*")
public SymDimVector get_base_ptrs()

_unsafe_set_arg_strides

public void _unsafe_set_arg_strides(int arg,
                                    @ByVal
                                    LongArrayRef strides)

_unsafe_set_arg_strides

public void _unsafe_set_arg_strides(int arg,
                                    @ByVal @Cast(value={"int64_t*","c10::ArrayRef<int64_t>","std::vector<int64_t>&"}) @StdVector(value="int64_t")
                                    long... strides)

_unsafe_set_arg_data

public void _unsafe_set_arg_data(int arg,
                                 Pointer data)

can_use_32bit_indexing

@Cast(value="bool")
public boolean can_use_32bit_indexing()

true if the stride computation can use 32-bit arithmetic. Used by GPU kernels

with_32bit_indexing

@ByVal
public SplitUntil32Bit with_32bit_indexing()

An "iteratable" object that recursively splits this iterator into sub-iterators that can use 32-bit indexing.

should_accumulate

@Cast(value="bool")
public boolean should_accumulate()

If the kernel should accumulate into the output. Only relevant for CUDA reductions.

is_final_output

@Cast(value="bool")
public boolean is_final_output()

Whether this iterator produces the actual output, as opposed to something that will be accumulated further. Only relevant for CUDA reductions.

has_contiguous_first_dim

@Cast(value="bool")
public boolean has_contiguous_first_dim()

set_output_raw_strided

public void set_output_raw_strided(@Cast(value="int64_t")
                                   long output_idx,
                                   @ByVal
                                   LongArrayRef sizes,
                                   @ByVal
                                   LongArrayRef strides,
                                   @ByVal
                                   TensorOptions options,
                                   @ByVal
                                   DimnameArrayRef names)

Overrides:

set_output_raw_strided in class MetaBase

set_output_raw_strided

public void set_output_raw_strided(@Cast(value="int64_t")
                                   long output_idx,
                                   @ByVal @Cast(value={"int64_t*","c10::ArrayRef<int64_t>","std::vector<int64_t>&"}) @StdVector(value="int64_t")
                                   long[] sizes,
                                   @ByVal @Cast(value={"int64_t*","c10::ArrayRef<int64_t>","std::vector<int64_t>&"}) @StdVector(value="int64_t")
                                   long[] strides,
                                   @ByVal
                                   TensorOptions options,
                                   @ByVal
                                   DimnameVector names)

Overrides:

set_output_raw_strided in class MetaBase

build_binary_float_op

public void build_binary_float_op(@Const @ByRef
                                  TensorBase out,
                                  @Const @ByRef
                                  TensorBase a,
                                  @Const @ByRef
                                  TensorBase b)

build_borrowing_binary_float_op

public void build_borrowing_binary_float_op(@Const @ByRef
                                            TensorBase out,
                                            @Const @ByRef
                                            TensorBase a,
                                            @Const @ByRef
                                            TensorBase b)

build_binary_op

public void build_binary_op(@Const @ByRef
                            TensorBase out,
                            @Const @ByRef
                            TensorBase a,
                            @Const @ByRef
                            TensorBase b)

build_borrowing_binary_op

public void build_borrowing_binary_op(@Const @ByRef
                                      TensorBase out,
                                      @Const @ByRef
                                      TensorBase a,
                                      @Const @ByRef
                                      TensorBase b)

build_unary_float_op

public void build_unary_float_op(@Const @ByRef
                                 TensorBase out,
                                 @Const @ByRef
                                 TensorBase a)

build_borrowing_unary_float_op

public void build_borrowing_unary_float_op(@Const @ByRef
                                           TensorBase out,
                                           @Const @ByRef
                                           TensorBase a)

build_unary_op

public void build_unary_op(@Const @ByRef
                           TensorBase out,
                           @Const @ByRef
                           TensorBase a)

build_output_borrowing_argument_owning_unary_op

public void build_output_borrowing_argument_owning_unary_op(@Const @ByRef
                                                            TensorBase out,
                                                            @Const @ByRef
                                                            TensorBase a)

build_borrowing_unary_op

public void build_borrowing_unary_op(@Const @ByRef
                                     TensorBase out,
                                     @Const @ByRef
                                     TensorBase a)

build_borrowing_unary_force_boolean_op

public void build_borrowing_unary_force_boolean_op(@Const @ByRef
                                                   TensorBase out,
                                                   @Const @ByRef
                                                   TensorBase a)

build_comparison_op

public void build_comparison_op(@Const @ByRef
                                TensorBase out,
                                @Const @ByRef
                                TensorBase a,
                                @Const @ByRef
                                TensorBase b)

build_borrowing_comparison_op

public void build_borrowing_comparison_op(@Const @ByRef
                                          TensorBase out,
                                          @Const @ByRef
                                          TensorBase a,
                                          @Const @ByRef
                                          TensorBase b)

build_borrowing_except_last_argument_comparison_op

public void build_borrowing_except_last_argument_comparison_op(@Const @ByRef
                                                               TensorBase out,
                                                               @Const @ByRef
                                                               TensorBase a,
                                                               @Const @ByRef
                                                               TensorBase b)

build_ternary_op

public void build_ternary_op(@Const @ByRef
                             TensorBase out,
                             @Const @ByRef
                             TensorBase a,
                             @Const @ByRef
                             TensorBase b,
                             @Const @ByRef
                             TensorBase c)