| Class and Description |
|---|
| GraphVertex
A GraphVertex is a vertex in the computation graph type of neural network.
|
| Class and Description |
|---|
| GraphVertex
A GraphVertex is a vertex in the computation graph type of neural network.
|
| Class and Description |
|---|
| AttentionVertex
Implements Dot Product Attention using the given inputs.
|
| AttentionVertex.Builder |
| ElementWiseVertex
An ElementWiseVertex is used to combine the activations of two or more layer in an element-wise manner
For example, the activations may be combined by addition, subtraction, multiplication (product), average or by selecting the maximum. |
| ElementWiseVertex.Op |
| GraphVertex
A GraphVertex is a vertex in the computation graph type of neural network.
|
| L2NormalizeVertex
L2NormalizeVertex performs L2 normalization on a single input, along the specified dimensions.
|
| L2Vertex
L2Vertex calculates the L2 (Euclidean) least squares error of two inputs, on a per-example basis.
|
| MergeVertex
A MergeVertex is used to combine the activations of two or more layers/GraphVertex by means of concatenation/merging.
|
| PoolHelperVertex
Removes the first column and row from an input.
|
| ReshapeVertex
Adds the ability to reshape and flatten the tensor in the computation graph.
|
| ScaleVertex
A ScaleVertex is used to scale the size of activations of a single layer: this is simply multiplication by a
fixed scalar value
For example, ResNet activations can be scaled in repeating blocks to keep variance under control. |
| ShiftVertex
A ShiftVertex is used to shift the activations of a single layer.
|
| StackVertex
StackVertex allows for stacking of inputs so that they may be forwarded through a network.
|
| SubsetVertex
SubsetVertex is used to select a subset of the activations out of another GraphVertex.
|
| UnstackVertex
UnstackVertex allows for unstacking of inputs so that they may be forwarded through
a network.
|
| Class and Description |
|---|
| GraphVertex
A GraphVertex is a vertex in the computation graph type of neural network.
|
| Class and Description |
|---|
| GraphVertex
A GraphVertex is a vertex in the computation graph type of neural network.
|
| Class and Description |
|---|
| GraphVertex
A GraphVertex is a vertex in the computation graph type of neural network.
|
| Class and Description |
|---|
| ElementWiseVertex.Op |
| Class and Description |
|---|
| GraphVertex
A GraphVertex is a vertex in the computation graph type of neural network.
|
Copyright © 2020. All rights reserved.