DLPrimitives
Public Member Functions | List of all members
dlprim::core::Scale Class Reference

Scale tensor by factor inplace, if s==0 fills with zero so nan is not propagated of s==0. More...

#include <include/dlprim/core/common.hpp>

Public Member Functions

 Scale (Context &ctx, DataType dtype=float_data)
 
void enqueue (float s, Tensor &t, ExecutionContext const &ec)
 

Detailed Description

Scale tensor by factor inplace, if s==0 fills with zero so nan is not propagated of s==0.


The documentation for this class was generated from the following file: