#python #tensorflow #machine-learning #keras #pytorch
#питон #тензорный поток #машинное обучение #keras #пыторч
Вопрос:
Я хочу создать равный слой keras из этого источника:
Layer=torch.nn.Conv1d(in_features, out_features, 1)
Мой ввод имеет форму (Batch_size, канал, ширина)
Этот слой компилируется для:
Conv1d(10, 256, kernel_size=(1,), stride=(1,))
Автор: pytorch.
Как я могу выразить этот слой в tensorflow?
У меня до сих пор есть это:
layer1 = tf.keras.layers.Conv1D(in_features-out_features 1, kernel_size=1)
Но я не уверен, что это завещание — правильный подход.
Ответ №1:
В keras от tensorflow вы пишете что-то вроде:
layer1 = tf.keras.layers.Conv1D(filters=256, kernel_size=1)(layer0)
Комментарии:
1. значит, аргумент фильтра — это просто out_features?
2. Да, это количество фильтров в layer1 (номер канала)