Skip to content

Commit

Permalink
fix error
Browse files Browse the repository at this point in the history
  • Loading branch information
mganahl committed Jul 2, 2020
1 parent 3de6545 commit 2e57c0a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tensornetwork/backends/jax/jax_backend.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ def reshape(self, tensor: Tensor, shape: Tensor) -> Tensor:
return jnp.reshape(tensor, np.asarray(shape).astype(np.int32))

def transpose(self, tensor, perm=None):
return jnp.transpose(tensor, perm)
return jnp.transpose(tensor, perm)

def shape_concat(self, values: Tensor, axis: int) -> Tensor:
return np.concatenate(values, axis)
Expand Down

0 comments on commit 2e57c0a

Please sign in to comment.