Pythonいぬ

pythonを使った画像処理に関する記事を書いていきます

2020-02-16から1日間の記事一覧

Pytorch tensor と numpy ndarray の変換

Pytorch tensor から numpy ndarray への変換とその逆変換についてまとめる。単純にtorch.from_numpy(x)とx.detach().numpy()を覚えておけばよいので、その使い方を示しておく。 すぐ使いたい場合は以下 numpy to tensor x = torch.from_numpy(x.astype(np.f…