Functions
ngraph.opset7.ops Namespace Reference

Functions

Node einsum (List[Node] inputs, str equation)
 Return a node which performs Einsum operation. More...
 
Node gelu (Node data, str approximation_mode, Optional[str] name=None)
 Return a node which performs Gelu activation function. More...
 
Node roll (NodeInput data, NodeInput shift, NodeInput axes)
 Return a node which performs Roll operation. More...
 
Node gather (NodeInput data, NodeInput indices, NodeInput axis, Optional[int] batch_dims=0)
 Return a node which performs Gather. More...
 
Node dft (NodeInput data, NodeInput axes, Optional[NodeInput] signal_size=None)
 Return a node which performs DFT operation. More...
 
Node idft (NodeInput data, NodeInput axes, Optional[NodeInput] signal_size=None)
 Return a node which performs IDFT operation. More...
 

Function Documentation

◆ dft()

Node ngraph.opset7.ops.dft ( NodeInput  data,
NodeInput  axes,
Optional[NodeInput]   signal_size = None 
)

Return a node which performs DFT operation.

@param data: Tensor with transformed data.
@param axes: Tensor with axes to transform.
@param signal_size: Tensor specifying signal size with respect to axes from the input 'axes'.
@return: The new node which performs DFT operation on the input data tensor.

◆ einsum()

Node ngraph.opset7.ops.einsum ( List[Node]  inputs,
str   equation 
)

Return a node which performs Einsum operation.

@param inputs: The list of input nodes
@param equation: Einsum equation
@return: The new node performing Einsum operation on the inputs

◆ gather()

Node ngraph.opset7.ops.gather ( NodeInput  data,
NodeInput  indices,
NodeInput  axis,
Optional[int]   batch_dims = 0 
)

Return a node which performs Gather.

@param data:         N-D tensor with data for gathering
@param indices:      N-D tensor with indices by which data is gathered
@param axis:         axis along which elements are gathered
@param batch_dims:   number of batch dimensions
@return:             The new node which performs Gather

◆ gelu()

Node ngraph.opset7.ops.gelu ( Node  data,
str  approximation_mode,
Optional[str]   name = None 
)

Return a node which performs Gelu activation function.

@param data: The node with data tensor.
@param approximation_mode: defines which approximation to use ('tanh' or 'erf')
@param name: Optional output node name.
@return The new node performing a Gelu activation with the input tensor.

◆ idft()

Node ngraph.opset7.ops.idft ( NodeInput  data,
NodeInput  axes,
Optional[NodeInput]   signal_size = None 
)

Return a node which performs IDFT operation.

@param data: Tensor with transformed data.
@param axes: Tensor with axes to transform.
@param signal_size: Tensor specifying signal size with respect to axes from the input 'axes'.
@return: The new node which performs IDFT operation on the input data tensor.

◆ roll()

Node ngraph.opset7.ops.roll ( NodeInput  data,
NodeInput  shift,
NodeInput  axes 
)

Return a node which performs Roll operation.

@param data: The node with data tensor.
@param shift: The node with the tensor with numbers of places by which elements are shifted.
@param axes: The node with the tensor with axes along which elements are shifted.
@return The new node performing a Roll operation on the input tensor.