def get_board(state):
 planes = np.zeros((3, state.size, state.size))
 planes[0, :, :] = (state.board == state.current_player)
 planes[1, :, :] = (state.board == (- state.current_player))
 planes[2, :, :] = (state.board == go.EMPTY)
 return planes

def get_board ( state ) : planes = np . zeros ( ( 3 , state . size , state . size ) ) planes [ 0 , : , : ] = ( state . board == state . current_player ) planes [ 1 , : , : ] = ( state . board == ( - state . current_player ) ) planes [ 2 , : , : ] = ( state . board == go . EMPTY ) return planes
a feature encoding white black and empty on separate planes .