cellflow.networks.TokenAttentionPooling.put_variable

TokenAttentionPooling.put_variable(col, name, value)

Updates the value of the given variable if it is mutable, or an error otherwise.

Args:

col: the variable collection. name: the name of the variable. value: the new value of the variable.

Parameters: