Searched refs:input_feature (Results 1 – 1 of 1) sorted by relevance
99 input_feature = F.reshape(x, (tensor_shape[0] * tensor_shape[1], tensor_shape[2]))100 output = self.matmul(input_feature, self.weight)215 def construct(self, input_feature): argument217 input_feature = self.fc(input_feature)219 input_feature = self.dropout(input_feature)221 input_feature = self.activation(input_feature)222 output_feature = self.reduce_mean(input_feature, 1)297 def construct(self, input_feature, bias_mat): argument298 input_feature = self.in_drop(input_feature)300 feature = self.feature_transform(input_feature)[all …]