初期化インデックス配列の初期化を行います。GPU使用時はベクトル空間の次元dとresourceとflat_configを以下のように設定する必要があります。print("BUILD THE INDEX") res = faiss.StandardGpuResources() flat_config = faiss.GpuIndexFlatConfig() index = faiss.GpuIndexFlatL2(res, d, flat_config)...
のように0~9に対する確率10個セットがバッチサイズ個ある二次元配列になっているので(出力.size()をすると、(バッチサイズ, 10) と出てくるはずである)、引数でdim = 1と指定することで、バッチサイズ個ある確率10個セットそれぞれの中だけでソフトマックスを行います。 これでネッ...
このコードでは、cp.ones((200, 200))で初期化した二次元配列(画像)の全画素について、 「周辺8画素の平均をとる」という操作を行います。 当然結果はcp.ones((200, 200))になるはずです。 しかし、operationの最後の行で、input[y_idx, x_idx] += 10;によって入力をいじっています。 こ...