logo

플럭스-파이토치-텐서플로우 치트 시트 📂머신러닝

플럭스-파이토치-텐서플로우 치트 시트

개요

플럭스, 파이토치, 텐서플로우에서 같은 기능을 하는 코드를 정리한다.

플럭스에 대해서 다음과 같은 환경이라고 하자.

using Flux

파이토치에 대해서 다음과 같은 환경이라고 하자.

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

텐서플로우에 대해서 다음과 같은 환경이라고 하자.

import tensorflow as tf
from tensorflow import keras

1차원 텐서

줄리아Julia파이토치PyTorch텐서플로우TensorFlow
열 벡터column vector
[1 4 -1 2][1;4;-1;2]