updated documenation, refactored function

Author: atroiano

NAMESPACE

@@ -302,6 +302,7 @@ export(layer_activation_thresholded_relu)
 export(layer_activity_regularization)
 export(layer_add)
 export(layer_alpha_dropout)
+export(layer_attention)
 export(layer_average)
 export(layer_average_pooling_1d)
 export(layer_average_pooling_2d)

R/layers-attention.R R/layer-attention.RR/layers-attention.R renamed to R/layer-attention.R

@@ -13,9 +13,19 @@
 #' @family attention layers   
 #'         
 #' @export
-layer_attention <- function(inputs, use_scale=FALSE, causal = FALSE) {
-  
-  create_layer(tf$keras$layers$Attention,object = inputs,args = list(use_scale = use_scale, causal = causal) )
+layer_attention <- function(inputs,use_scale=FALSE, causal = FALSE, batch_size = NULL, dtype = NULL, 
+                                name = NULL, trainable = NULL, weights = NULL) {
+  create_layer(keras$layers$Attention, inputs, list(
+      use_scale = use_scale,
+      causal = causal,
+      batch_size = batch_size,
+      dtype = dtype, 
+      name = name,
+      trainable = trainable, 
+      weights = weights)
+    )
+    
+    
 }