Interface FunctionElementWiseMult<T extends Tensor>

All Superinterfaces:
Function<T>
All Known Implementing Classes:
FunctionElementWiseMult_F32, FunctionElementWiseMult_F64

public interface FunctionElementWiseMult<T extends Tensor> extends Function<T>
Multiplies each element in a tensor by the same value.
  • Method Details

    • forward

      void 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
       
      Specified by:
      forward in interface Function<T extends Tensor>
      Parameters:
      input - Input tensor of any shape.
      output - Output tensor. Same shape as input tensor.