Sha256: 3652f5b75cc06309dc6d9dc95adf48b1b2fd1e83c24962dd6fbce1500581065a
Contents?: true
Size: 364 Bytes
Versions: 9
Compression:
Stored size: 364 Bytes
Contents
TensorStream::OpMaker.define_operation :tan do |op|
op.what_it_does "Computes tan of input element-wise."
op.parameter :input_a, "tensor X", validate: 'FLOATING_POINT_TYPES'
op.option :name, "Optional name", :nil
op.define_gradient do |grad, node, params|
secx = ts.reciprocal(ts.cos(params[0]))
secx2 = ts.square(secx)
grad * secx2
end
end
Version data entries
9 entries across 9 versions & 1 rubygems