25 lines
552 B
Python
25 lines
552 B
Python
|
"""Use cases for testing matmul (@)
|
||
|
|
"""
|
||
|
|
def matmul_usecase(x, y):
|
||
|
|
return x @ y
|
||
|
|
|
||
|
|
def imatmul_usecase(x, y):
|
||
|
|
x @= y
|
||
|
|
return x
|
||
|
|
|
||
|
|
class DumbMatrix(object):
|
||
|
|
|
||
|
|
def __init__(self, value):
|
||
|
|
self.value = value
|
||
|
|
|
||
|
|
def __matmul__(self, other):
|
||
|
|
if isinstance(other, DumbMatrix):
|
||
|
|
return DumbMatrix(self.value * other.value)
|
||
|
|
return NotImplemented
|
||
|
|
|
||
|
|
def __imatmul__(self, other):
|
||
|
|
if isinstance(other, DumbMatrix):
|
||
|
|
self.value *= other.value
|
||
|
|
return self
|
||
|
|
return NotImplemented
|