Skip to content

Commit

Permalink
expose exp and log on a tensor
Browse files Browse the repository at this point in the history
  • Loading branch information
darrenjw committed Mar 18, 2023
1 parent 2808124 commit f79ce0c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion core/src/main/scala/torch/Tensor.scala
Original file line number Diff line number Diff line change
Expand Up @@ -282,7 +282,7 @@ sealed abstract class Tensor[D <: DType]( /* private[torch] */ val native: pyto
/** True if `other` has the same size and elements as this tensor, false otherwise. */
def equal(other: Tensor[D]): Boolean = native.equal(other.native)

/** Returns the tensor with element exponentiated. */
/** Returns the tensor with elements exponentiated. */
def exp: Tensor[D] = Tensor(native.exp())

def flatten: Tensor[D] = Tensor(native.flatten())
Expand Down

0 comments on commit f79ce0c

Please sign in to comment.