import tensorflow as tf

    def build_network(self, u, v, w, u_mag, v_mag, w_mag, low_resblock=8, hi_resblock=4, channel_nr=64):
        channel_nr = 64

        speed = (u ** 2 + v ** 2 + w ** 2) ** 0.5
        mag = (u_mag ** 2 + v_mag ** 2 + w_mag ** 2) ** 0.5
        pcmr = mag * speed

        phase = tf.keras.layers.concatenate([u,v,w])
        pc    = tf.keras.layers.concatenate([pcmr, mag, speed])
        
        pc = conv3d(pc,3,channel_nr, 'SYMMETRIC', 'relu')
        pc = conv3d(pc,3,channel_nr, 'SYMMETRIC', 'relu')

        phase = conv3d(phase,3,channel_nr, 'SYMMETRIC', 'relu')
