([input_dim, output_dim]))y = tf.nn.softmax(a)
因此,变量y的维度为batch_size by output_dim我想要计算批次中每个样品相对于a的y雅可比矩阵,它的维度是batch_size by output_dim by output_dim。现在,在数学上,当i !我想知道如何计算softmax相对于其在TensorFlow中的输入<em
我试图复制这个项目的代码:batch_size <- 100000
model <- make_model(n_rows = dim(train_data_keras/site-packages/tensorflow_probability/python/bijectors/composition.py", line 552, in _inverse_log_det_jacobian/lib/pyth