Printing of three dimensional tensor

print_tensor.py

# Assumption tensor == np.array
def print_tensor(tensor):
    for i3 in range(0, tensor.shape[0]):
        for i1 in range(0, tensor.shape[2]):
            string_line = ""
            for i2 in range(0, tensor.shape[1]):
                string_line += " {:3.2f} ".format(tensor[i3][i2][i1]) 
            print(string_line)
        print("------------------------------------")