General layer structure. More...

Data Fields
ailayer_t	base
	Inherited field members from general ailayer struct.

Layer configuration
Required configuration parameters for the layer These fields have to be configured by the user before calling the initializer function.
uint8_t	output_dim
	Output dimension count (length of output shape array).

uint8_t	infer_axis
	Specifies the axis that is inferred from the elements of the tensor and remaining axes (Set to zero if not used).

uint16_t *	output_shape
	Target shape for reshape operation.

Math functions
Required data type specific math functions
void(*	reshape )(const aitensor_t x, aitensor_t result)
	Optional math function: Reshape. More...

Detailed Description

General layer structure.

Field Documentation

void(* reshape) (const aitensor_t *x, aitensor_t *result)

Optional math function: Reshape.

Optional math function that reshapes the tensor x (may be due to copying or just changing the shape).
Set to zero if not required.

Parameters

x	N-dimensional tensor (input)
result	M-dimensional tensor (output)

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