org.bytedeco.tensorflow

Class MatrixDiagPart

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.tensorflow.MatrixDiagPart

All Implemented Interfaces:

AutoCloseable

@Namespace(value="tensorflow::ops")
 @NoOffset
 @Properties(inherit=tensorflow.class)
public class MatrixDiagPart
extends Pointer

Returns the batched diagonal part of a batched tensor. This operation returns a tensor with the diagonal part of the batched input. The diagonal part is computed as follows: Assume input has k dimensions [I, J, K, ..., M, N], then the output is a tensor of rank k - 1 with dimensions [I, J, K, ..., min(M, N)] where: diagonal[i, j, k, ..., n] = input[i, j, k, ..., n, n]. The input must be at least a matrix. For example:

  # 'input' is [[[1, 0, 0, 0]
                 [0, 2, 0, 0]
                 [0, 0, 3, 0]
                 [0, 0, 0, 4]],
                [[5, 0, 0, 0]
                 [0, 6, 0, 0]
                 [0, 0, 7, 0]
                 [0, 0, 0, 8]]]
 
  and input.shape = (2, 4, 4)
 
  tf.matrix_diag_part(input) ==> [[1, 2, 3, 4], [5, 6, 7, 8]]
 
  which has shape (2, 4)
  

Arguments: * scope: A Scope object * input: Rank k tensor where k >= 2. Returns: * Output: The extracted diagonal(s) having shape diagonal.shape = input.shape[:-2] + [min(input.shape[-2:])].

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
MatrixDiagPart(Pointer p) Pointer cast constructor.
MatrixDiagPart(Scope scope, Input input)

Method Summary

Modifier and Type	Method and Description
Input	asInput()
Output	asOutput()
Output	diagonal()
MatrixDiagPart	diagonal(Output setter)
Node	node()
Operation	operation()
MatrixDiagPart	operation(Operation setter)

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

MatrixDiagPart

public MatrixDiagPart(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

MatrixDiagPart

public MatrixDiagPart(@Const @ByRef
                      Scope scope,
                      @ByVal
                      Input input)

Method Detail

asOutput

@ByVal
 @Name(value="operator tensorflow::Output")
public Output asOutput()

asInput

@ByVal
 @Name(value="operator tensorflow::Input")
public Input asInput()

node

public Node node()

operation

@ByRef
public Operation operation()

operation

public MatrixDiagPart operation(Operation setter)

diagonal

@ByRef
public Output diagonal()

diagonal

public MatrixDiagPart diagonal(Output setter)