def make_d4_z2_indices(ksize): assert ksize % 2 == 1 # TODO x = np.random.randn(1, ksize, ksize) f = Z2FuncArray(v=x) uv = f.left_translation_indices(D4.flatten()[:, None, None, None]) mr = np.zeros(uv.shape[:-1] + (1,)) mruv = np.c_[mr, uv] return mruv.astype('int32')
def make_d4_z2_indices(ksize): assert ksize % 2 == 1 # TODO x = np.random.randn(1, ksize, ksize) f = Z2FuncArray(v=x) uv = f.left_translation_indices(D4.flatten()[:, None, None, None]) mr = np.zeros(uv.shape[:-1] + (1, )) mruv = np.c_[mr, uv] return mruv.astype('int32')
def make_d4_z2_indices(ksize): assert ksize % 2 == 1 # TODO x = np.random.randn(1, ksize, ksize) # 1 relates to stabilizer size of input channel f = Z2FuncArray(v=x) # input channel related uv = f.left_translation_indices(D4.flatten()[:, None, None, None]) # this adds the output channel stabilizer # output channel related mr = np.zeros(uv.shape[:-1] + (1,)) mruv = np.c_[mr, uv] return mruv.astype('int32')
def make_d4_p4m_indices(ksize): assert ksize % 2 == 1 # TODO x = np.random.randn(8, ksize, ksize) # 8 3 3 f = P4MFuncArray(v=x) li = f.left_translation_indices(D4.flatten()[:, None, None, None]) #8 8 3 3 3 #print(li.astype('int32')) #print('li.shape', li.shape) return li.astype('int32')
def make_d4_p4m_indices(ksize): assert ksize % 2 == 1 # TODO x = np.random.randn(8, ksize, ksize) f = P4MFuncArray(v=x) li = f.left_translation_indices(D4.flatten()[:, None, None, None]) return li.astype('int32')
def make_d4_p4m_indices(ksize): assert ksize % 2 == 1 # TODO x = np.random.randn(8, ksize, ksize) # input channel related f = P4MFuncArray(v=x) # input channel related li = f.left_translation_indices(D4.flatten()[:, None, None, None]) # output channel related return li.astype('int32')