def check_type_forward(self, in_types): type_check.expect(in_types.size() == 1) a_type, = in_types a_type = matmul._convert_type(a_type) type_check.expect(a_type.dtype.kind == 'f') # Only a minibatch of 2D array shapes allowed. type_check.expect(a_type.ndim == 3) # Matrix inversion only allowed for square matrices # so assert the last two dimensions are equal. type_check.expect(a_type.shape[-1] == a_type.shape[-2])
def check_type_forward(self, in_types): type_check.expect(in_types.size() == 1) a_type, = in_types a_type = matmul._convert_type(a_type) type_check.expect(a_type.dtype.kind == 'f') # Only a minibatch of 2D array shapes allowed. type_check.expect(a_type.ndim == 3) # Matrix inversion only allowed for square matrices # so assert the last two dimensions are equal. type_check.expect(a_type.shape[-1] == a_type.shape[-2])
def check_type_forward(self, in_types): type_check.expect(in_types.size() == 1) a_type, = in_types type_check.expect(a_type.dtype.kind == 'f', ) _check_ndim(a_type) a_type = _convert_type(a_type) a_idx = _get_check_index(self.transa, False)
def check_type_forward(self, in_types): type_check.expect(in_types.size() == 1) a_type = self.value b_type = in_types[0] type_check.expect(b_type.dtype.kind == "f") _matmul._check_ndim(b_type) b_type = _matmul._convert_type(b_type) a_idx = _matmul._get_check_index(False, False) b_idx = _matmul._get_check_index(False, True) type_check.expect(a_type.shape[a_idx] == b_type.shape[b_idx])
def check_type_forward(self, in_types): type_check.expect(in_types.size() == 1) a_type = self.value b_type = in_types[0] type_check.expect(b_type.dtype.kind == 'f') _matmul._check_ndim(b_type) b_type = _matmul._convert_type(b_type) a_idx = _matmul._get_check_index(False, False) b_idx = _matmul._get_check_index(False, True) type_check.expect(a_type.shape[a_idx] == b_type.shape[b_idx])
def check_type_forward(self, in_types): type_check.expect(in_types.size() == 1) a_type, = in_types type_check.expect( a_type.dtype == numpy.float32 # , # self.b.dtype == numpy.float32 ) _check_ndim(a_type, lower=2, upper=3) # _check_ndim(b_type, lower=2, upper=3) a_type = _convert_type(a_type, vector_ndim=2) # b_type = _convert_type(b_type, vector_ndim=2) a_idx = _get_check_index(self.transa, False, row_idx=1, col_idx=2)