Numpyの数値の取り出し

Question

ChainerやPyTorch のオブジェクト変数のdataを参照すると、属性まで取得されますが、数値だけ取り出すことはできますか。

!pip install Chainer

import numpy as np
import chainer

class Model(chainer.Chain):

    def __init__(self ):
        super(Model, self).__init__()
        with self.init_scope():
            self.l1 = chainer.links.Linear(None, 1) 

    def __call__(self, x):
        return self.l1(x)

model = Model()
x = np.array([[0.1, 0.2, 0.3 ]],dtype='float32')
y = model(x)
y.data

出力は
array([[0.5010003]], dtype=float32)
です。
このうち、0.5010003 だけ数値として取得したいです。

Answer 1

こんにちは。
array([[-0.15829726]], dtype=float32)は
numpy.ndarrayで、float型しか入らない2次元配列です。

配列の中身はインデックスで取り出せます。
2次元配列（配列の配列）なので下のどちらかの方法で取り出しましょう。

y.data[0][0]
y.data[0, 0]

参考：NumPy 配列の基礎 — 機械学習の Python との出会い

Comments

[teefpc]

できました！ありがとうございました。

Answer 2

y.data→y.data[0][0]で数値のみの出力ができます。

Comments

[teefpc]

できました！ありがとうございました。