EinsumDense¶
Layer that uses einsum.
Abstract Signature:
EinsumDense(equation: str, output_shape: int | Tuple[int, ...], bias_axes: str)
PyTorch
API:
—Strategy: Custom / Partial
Layer that uses einsum.
Abstract Signature:
EinsumDense(equation: str, output_shape: int | Tuple[int, ...], bias_axes: str)
—