ls = np.log(ys[np.arange(N * T), ts])
がピンと来ないので実験。
>>> ys = np.array([[1,2],[3,4],[5,6],[7,8],[9,10]]) >>> ts = np.array([0,0,1,1,0]) >>> ys[np.arange(5), ts] array([1, 3, 6, 8, 9])
なるほど・・・。
ls = np.log(ys[np.arange(N * T), ts])
がピンと来ないので実験。
>>> ys = np.array([[1,2],[3,4],[5,6],[7,8],[9,10]]) >>> ts = np.array([0,0,1,1,0]) >>> ys[np.arange(5), ts] array([1, 3, 6, 8, 9])
なるほど・・・。