TensorFlow 2 и Keras:

Не знаете, зачем вам нужен w, h. Если эти значения действительно необходимы и означают, что нужно читать только указанное количество строк и столбцов, вы можете попробовать следующее:

output = []
with open(r'c:\file.txt', 'r') as f:
    w, h  = map(int, f.readline().split())
    tmp = []
    for i, line in enumerate(f):
        if i == h:
            break
        tmp.append(map(int, line.split()[:w]))
    output.append(tmp)