📂機械学習

Flux-PyTorch-TensorFlowチートシート

概要

Flux、PyTorch、TensorFlowで同じ機能をするコードを整理します。

• Julia-MATLAB-Python-R チートシート

Fluxについて次のような環境とします。

using Flux

PyTorchについて次のような環境とします。

import torch
import torch.nn as nn
import torch.nn.functional as F

TensorFlowについて次のような環境とします。

import tensorflow as tf
from tensorflow import keras

1次元テンソル

줄리아Julia	파이토치PyTorch	텐서플로우TensorFlow
列ベクトルcolumn vector
[1 4 -1 2]	[1;4;-1;2]