class nbla::TransformUnary

template<typename T, typename UnaryOp, typename ...Args>
class TransformUnary : public nbla::BaseTransformUnary<Args...>

Public Functions

inline virtual vector<dtypes> in_types()

Get input dtypes.

Last in_type will be used repeatedly if size of in_types is smaller than size of inputs

inline virtual vector<dtypes> out_types()

Get output dtypes.

Last out_type will be used repeatedly if size of out_types is smaller than size of outputs

inline virtual vector<string> allowed_array_classes()

Get array classes that are allowed to be specified by Context.