towhee.hub.builtin.operators.tensor_like.tensor_matmul

class towhee.hub.builtin.operators.tensor_like.tensor_matmul(trans=None)[source]

Bases: object

Matrix multiplication.

Examples: >>> import numpy as np >>> from towhee import DataFrame, Entity >>> from towhee.types.tensor_array import TensorArray >>> df = DataFrame([Entity(a = np.ones([2, 1]), b = np.ones([1, 2])) for _ in range(3)]) >>> df.tensor_matmul[(‘a’, ‘b’), ‘c’]().to_list() [<Entity dict_keys([‘a’, ‘b’, ‘c’])>, <Entity dict_keys([‘a’, ‘b’, ‘c’])>, <Entity dict_keys([‘a’, ‘b’, ‘c’])>]

>>> df.to_column()
>>> df.tensor_matmul[('a', 'b'), 'd']().to_list()
[<EntityView dict_keys(['a', 'b', 'c', 'd'])>, <EntityView dict_keys(['a', 'b', 'c', 'd'])>, <EntityView dict_keys(['a', 'b', 'c', 'd'])>]df

Methods

Attributes

metainfo

__call__(*args, **kws)

Call self as a function.

__init__(trans=None)[source]