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.