public interface FunctionElementWiseMult<T extends Tensor> extends Function<T>
| Modifier and Type | Method and Description |
|---|---|
void |
forward(T input,
T output)
Performs scalar multiplication on each element in the input tensor.
|
getOutputShape, getParameters, getParameterShapes, getTensorType, initialize, setParametersvoid forward(T input, T output)
Performs scalar multiplication on each element in the input tensor.
Summary Table ------------------------------------------------- Input shape = (N, d[i], ... , d[k]) ------------------------------------------------- N = Size of mini-batch d[i] = length of a dimension