chainer.dataset.to_device

chainer.dataset.to_device(device, x)[source]

与えられたデバイスにデータを送信する。

 

このメソッドは与えられたデバイスの与えられた配列へデータを送ります。このメソッドは concat_examples()で用いられます。また、このメソッドはカスタム変換メソッドUpdater や Extension でStandardUpdater やEvaluatorのように用いられるカスタム変換メソッドとして利用することも可能です。

 

chainer.dataset.concat_examples()もご覧ください。

 

Parameters:
  • device (int or None) – 配列が送られるデバイス。この値が負の場合配列はCPUへ怒られます。正の場合、配列は与えられたIDとともにGPUへ送られます。 Noneの場合、配列はもとにあったデバイスに残されたままになります。
  • x (numpy.ndarray or cupy.ndarray) – 送信する配列
Returns:

変換された配列