package tf
- Alphabetic
- By Inheritance
- tf
- AnyRef
- Any
- Hide All
- Show All
- Public
- All
Type Members
- class BiasAdd[T] extends AbstractModule[Table, Tensor[T], T]
-
sealed abstract
class
ControlOps[T] extends Operation[Activity, Activity, T]
Control flow related operations
Control flow related operations
- T
Numeric type of parameter(e.g. weight, bias). Only support float/double now
-
abstract
class
IdentityControl[T] extends ControlOps[T]
Control flow related operations, and they just pass the input without modifying them
Control flow related operations, and they just pass the input without modifying them
- T
Numeric type of parameter(e.g. weight, bias). Only support float/double now
-
class
Log1p[T, D] extends AbstractModule[Tensor[D], Tensor[D], T]
The Log module applies a log transformation to the input data
The Log module applies a log transformation to the input data
- Annotations
- @SerialVersionUID()
-
sealed
class
MergeControlNode[T] extends Node[T]
A wrapper of node for merge operation.
A wrapper of node for merge operation.
- T
element type
- class SqrtGrad[T, D] extends Operation[Table, Tensor[D], T]
-
sealed
class
SwitchControlNode[T] extends Node[T]
A wrapper of node for switch operation.
A wrapper of node for switch operation. Make code easy to read.
- T
element type
-
class
TensorModuleWrapper[T, D] extends AbstractModule[Tensor[D], Tensor[D], T]
The AbstractModule[Tensor[D], Tensor[D], T] that takes a TensorModule[D] to enable the computation of D typed data in a module with numeric type of T.