NumpyからTorch Tensorへの変換

NumpyのndarrayからTorch Tensorへの変換はfrom_numpyを使って以下のように行います。

x_torch = torch.from_numpy(x_numpy)

適当にnumpy.ndarrayを作って変換してみたサンプルは以下になります。

参考文献

コメント

タイトルとURLをコピーしました