入力を３０、出力を３の全結合層を定義したい

Question

入力を３０、出力を３の全結合層を定義したいのですが、書き方がわかりません。
Chainerだと、

L.Linear(30, 3)

のように書けると思います。tensorflowでの書き方がわかりません。
初心者です、よろしくお願い致します。

Answer 1

Tensorflow の Keras レイヤーを使うといいかと思います。

import tensorflow as tf
from tensorflow.keras.layers import Dense
from tensorflow.keras.models import Sequential

model = Sequential()
model.add(Dense(3, input_dim=30, activation='relu'))  # 入力30、出力3の全結合層
model.add(Dense(5, activation='relu'))  # 出力5の全結合層、入力数は前の層から自動的に決まるので、明示的に指定しなくてよい
model.summary()

Comments

[taiyo2017]

ありがとうございます！！