「Pythonで動かして学ぶ！ 新しい機械学習の教科書」を読んでいて、意味がわからない場所があるので教えて下さい。

Question

前提・実現したいこと

ここに質問の内容を詳しく書いてください。
「Pythonで動かして学ぶ！ 新しい機械学習の教科書」を読んでいて、
Pythonでソフトマックス関数を書いていますが、意味がわからない場所があるので、よろしければ教えて下さい。

発生している問題・エラーメッセージ

エラーはありません。

該当のソースコード

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
%matplotlib inline

def softmax(x0, x1, x2):
    u = np.exp(x0) + np.exp(x1) + np.exp(x2)
    return np.exp(x0) / u, np.exp(x1) / u, np.exp(x2) / u

xn = 20
x0 = np.linspace(-4, 4, xn)
x1 = np.linspace(-4, 4, xn)

y = np.zeros((xn, xn, 3))
for i0 in range(xn):
    for i1 in range(xn):
        y[i1, i0, :] = softmax(x0[i0], x1[i1], 1)
        
xx0, xx1 = np.meshgrid(x0, x1)
plt.figure(figsize=(8, 3))
for i in range(2):
    ax = plt.subplot(1, 2, i+1, projection='3d')
    ax.plot_surface(xx0, xx1, y[:, :, i], 
                    rstride=1,  cstride=1, alpha=0.3, 
                    color='blue', edgecolor='black')
    ax.set_xlabel('$x_0$', fontsize=14)
    ax.set_ylabel('$x_1$', fontsize=14)
    ax.view_init(40, -125)

試したこと

y[i1, i0, :]とy[:, :, i]が何をやっているか意味がわかりません。
あとrange(2)ではうまくいきましたが、range(3)ではうまくいきませんでした。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

y[i1, i0, :]とy[:, :, i]が何をやっているか意味がわかりません。

何の説明も無くそういう記述が出てきたのなら、その本はPythonを知っている人向けの本なのでしょう。
であれば、まずPythonに入門しましょう。もしくは、Pythonを知らない人向けの本を使うか。

リスト（配列）の添え字に、前後に値のない単独の:があれば、それはその軸での全部を表します。
a[:,1]だと2次元配列の「1列目」（つまり2番目の列）、a[0,:]だと`「0行目」（つまり最初の行）を表します。