Control flow related operations
Control flow related operations
Numeric type of parameter(e.g. weight, bias). Only support float/double now
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
Numeric type of parameter(e.g. weight, bias). Only support float/double now
The Log module applies a log transformation to the input data
The Log module applies a log transformation to the input data
A wrapper of node for merge operation.
A wrapper of node for merge operation.
element type
A wrapper of node for switch operation.
A wrapper of node for switch operation. Make code easy to read.
element type
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.